no load model print out

supernnova · Apr 30, 2020 · b4c7861 · b4c7861
1 parent f9f647f
commit b4c7861
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/supernnova/utils/training_utils.py b/supernnova/utils/training_utils.py
@@ -299,7 +299,8 @@ def get_model(settings, input_size):
 
     rnn = rnn(input_size, settings)
 
-    print(rnn)
+    if not settings.no_dump:
+        print(rnn)
 
     return rnn