Load model parameters only when specified in the config

2 files changed, 3 insertions, 1 deletions

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
                                ),
                 ]