Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/TharinduDR/TransQuest.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorTharinduDR <rhtdranasinghe@gmail.com>2020-06-30 18:15:22 +0300
committerTharinduDR <rhtdranasinghe@gmail.com>2020-06-30 18:15:22 +0300
commit7abb497bfa48fe76571d4c7ac61cbb6cb94293d8 (patch)
treedeb865d42f2f34aaf57b8ef9602781342ce3310f
parent37d143602beb6101d98b6216ed51fbac37ab9c2e (diff)
030: Adding multilingual experiments
-rwxr-xr-xexamples/wmt_2020/ro_en/siamese_trans_quest.py8
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)