diff --git a/sm-model/main.py b/sm-model/main.py index 140bdf7..2f6cefb 100644 --- a/sm-model/main.py +++ b/sm-model/main.py @@ -126,7 +126,7 @@ if __name__ == "__main__": if args.debugSingleBatch: sys.exit(0) dev_accuracy, dev_scores = trainer.test(args.dataset_folder, 'clean-dev', args.batch_size, cache_file) dev_map, dev_mrr = compute_map_mrr(args.dataset_folder, 'clean-dev', dev_scores) - logger.info("MAP {}, MRR {}".format(dev_map, dev_mrr)) + logger.info("MAP {}\nMRR {}".format(dev_map, dev_mrr)) if np.fabs(dev_map - best_map) > 1e-3: best_model = i @@ -148,6 +148,6 @@ if __name__ == "__main__": logger.info('Test set accuracy = {:.4f}'.format(test_accuracy)) map, mrr = compute_map_mrr(args.dataset_folder, 'clean-test', test_scores) - logger.info("MAP {}, MRR {}".format(map, mrr)) + logger.info("MAP {}\nMRR {}".format(map, mrr)) diff --git a/sm-model/train.py b/sm-model/train.py index 3bd2ffc..3ddab13 100644 --- a/sm-model/train.py +++ b/sm-model/train.py @@ -150,6 +150,8 @@ class Trainer(object): def train(self, dataset_folder, set_folder, batch_size, word_vectors_cache_file, debugSingleBatch): + train_start_time = time.time() + # read in training data questions, sentences, labels, vocab, maxlen_q, maxlen_s, ext_feats = \ utils.read_in_dataset(dataset_folder, set_folder) @@ -193,6 +195,7 @@ class Trainer(object): logger.info('train_loss = {:.4f}'.format( train_loss/num_batches )) + logger.info('training time = {:.3f} seconds'.format(time.time() - train_start_time)) return train_correct/num_batches