wassname/Castor
1
0
Fork 0
mirror of https://github.com/wassname/Castor.git synced 2026-09-09 11:13:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

important slight stopping criterion change

Browse Source
This commit is contained in:
Gaurav Baruah
2017-03-28 15:03:22 -04:00
parent b2c82f7e08
commit 67a93a736f
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sm-model/main.py
+2 -1
View File
@@ -116,6 +116,7 @@ if __name__ == "__main__":
torch.set_num_threads(args.num_threads)
trainer = Trainer(net, args.eta, args.mom, args.no_loss_reg)
logger.info("Loading input data...")
trainer.load_input_data(args.dataset_folder, cache_file, 'train', 'clean-dev', 'clean-test')
best_map = 0.0
@@ -130,7 +131,7 @@ if __name__ == "__main__":
dev_map, dev_mrr = compute_map_mrr(args.dataset_folder, 'clean-dev', dev_scores)
logger.info("------- MAP {}, MRR {}".format(dev_map, dev_mrr))
if np.fabs(dev_map - best_map) > 1e-3:
if dev_map - best_map > 1e-3: # new map is better than best map
best_model = i
best_map = dev_map
QAModel.save(net, args.dataset_folder, args.model_fname)