diff options
Diffstat (limited to 'examples/sentence_level/wmt_2020/et_en/siamesetransquest.py')
-rw-r--r-- | examples/sentence_level/wmt_2020/et_en/siamesetransquest.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/examples/sentence_level/wmt_2020/et_en/siamesetransquest.py b/examples/sentence_level/wmt_2020/et_en/siamesetransquest.py index 2444a75..f668a97 100644 --- a/examples/sentence_level/wmt_2020/et_en/siamesetransquest.py +++ b/examples/sentence_level/wmt_2020/et_en/siamesetransquest.py @@ -61,7 +61,7 @@ if siamesetransquest_config["evaluate_during_training"]: shutil.rmtree(siamesetransquest_config['cache_dir']) train_df, eval_df = train_test_split(train, test_size=0.1, random_state=SEED * i) - model = SiameseTransQuestModel(MODEL_NAME) + model = SiameseTransQuestModel(MODEL_NAME, args=siamesetransquest_config) model.train_model(train_df, eval_df) model = SiameseTransQuestModel(siamesetransquest_config['best_model_dir']) |