From 32b078f28add3d22529e55aeac6674d924e9b510 Mon Sep 17 00:00:00 2001 From: AdeB Date: Wed, 24 Jun 2015 15:28:04 -0400 Subject: Load model parameters only when specified in the config --- config/memory_network_adeb.py | 2 ++ train.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/config/memory_network_adeb.py b/config/memory_network_adeb.py index 1d7dc5d..a9d6fef 100644 --- a/config/memory_network_adeb.py +++ b/config/memory_network_adeb.py @@ -44,3 +44,5 @@ num_cuts = 1000 train_candidate_size = 1000 valid_candidate_size = 10000 + +load_model = False diff --git a/train.py b/train.py index 83317c9..17d5789 100755 --- a/train.py +++ b/train.py @@ -153,7 +153,7 @@ if __name__ == "__main__": Printing(every_n_batches=1000), SaveLoadParams(dump_path, cg, - before_training=True, # before training -> load params + before_training=config.load_model, # before training -> load params every_n_batches=1000, # every N batches -> save params ), ] -- cgit v1.2.3