taxi

commit 32b078f28add3d22529e55aeac6674d924e9b510
parent 88cdc3f8047a05bc5971eaa915ca6626f89a3e78
Author: AdeB <adbrebs@gmail.com>
Date:   Wed, 24 Jun 2015 15:28:04 -0400

Load model parameters only when specified in the config

Diffstat:
M
config/memory_network_adeb.py
 | 
2
++
M
train.py
 | 
2
+-

2 files changed, 3 insertions(+), 1 deletion(-)
diff --git 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
@@ -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
                                ),
                 ]