diff options
author | TharinduDR <rhtdranasinghe@gmail.com> | 2020-06-30 18:15:22 +0300 |
---|---|---|
committer | TharinduDR <rhtdranasinghe@gmail.com> | 2020-06-30 18:15:22 +0300 |
commit | 7abb497bfa48fe76571d4c7ac61cbb6cb94293d8 (patch) | |
tree | deb865d42f2f34aaf57b8ef9602781342ce3310f | |
parent | 37d143602beb6101d98b6216ed51fbac37ab9c2e (diff) |
030: Adding multilingual experiments
-rwxr-xr-x | examples/wmt_2020/ro_en/siamese_trans_quest.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/examples/wmt_2020/ro_en/siamese_trans_quest.py b/examples/wmt_2020/ro_en/siamese_trans_quest.py index 18007d3..4e1602c 100755 --- a/examples/wmt_2020/ro_en/siamese_trans_quest.py +++ b/examples/wmt_2020/ro_en/siamese_trans_quest.py @@ -106,7 +106,7 @@ if siamese_transformer_config["evaluate_during_training"]: len(train_data) * siamese_transformer_config["num_train_epochs"] / siamese_transformer_config[ 'train_batch_size'] * 0.1) - start = time.start() + start = time.time() model.fit(train_objectives=[(train_dataloader, train_loss)], evaluator=evaluator, epochs=siamese_transformer_config['num_train_epochs'], @@ -116,7 +116,7 @@ if siamese_transformer_config["evaluate_during_training"]: 'correct_bias': False}, warmup_steps=warmup_steps, output_path=siamese_transformer_config['best_model_dir']) - end = time.end() + end = time.time() print("Training time") print(end - start) @@ -125,11 +125,11 @@ if siamese_transformer_config["evaluate_during_training"]: dev_data = SentencesDataset(examples=sts_reader.get_examples("dev.tsv"), model=model) dev_dataloader = DataLoader(dev_data, shuffle=False, batch_size=8) evaluator = EmbeddingSimilarityEvaluator(dev_dataloader) - start = time.start() + start = time.time() model.evaluate(evaluator, result_path=os.path.join(siamese_transformer_config['cache_dir'], "dev_result.txt")) - end = time.end() + end = time.time() print("Testing time") print(end - start) |