diff --git a/classifier_with_adapter.py b/classifier_with_adapter.py index a8377675e4dedcfcb5fe45b98c02fd768e5a14f8..9dca050d05fc45d7622c42057e4ee4e3fafb43fa 100644 --- a/classifier_with_adapter.py +++ b/classifier_with_adapter.py @@ -29,7 +29,7 @@ mappings, inv_mappings = open_mappings(args.mappings_file) substitutions_file = 'mappings/substitutions.txt' tokenizer = AutoTokenizer.from_pretrained(args.transformer_model) -set_seed(42) +set_seed(578) print('Train classifier with adapter\n') print('Adapter name:', adapter_name) @@ -153,7 +153,7 @@ if args.freeze_layers != '': # Start the training 🚀 # =============================== -print('Start training...') +# print('Start training...') trainer.train() # Dev results