From 67a93a736fe91670079941b5b32296aee2037a18 Mon Sep 17 00:00:00 2001 From: Gaurav Baruah Date: Tue, 28 Mar 2017 15:03:22 -0400 Subject: [PATCH] important slight stopping criterion change --- sm-model/main.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/sm-model/main.py b/sm-model/main.py index 0e963ff..1aa6d91 100644 --- a/sm-model/main.py +++ b/sm-model/main.py @@ -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)