Made the model filenames more explicit.

[pysvrt.git] / cnn-svrt.py

diff --git a/cnn-svrt.py b/cnn-svrt.py
index d44de2a..cfef09d 100755 (executable)
--- a/cnn-svrt.py
+++ b/cnn-svrt.py
@@ -284,8 +284,8 @@ for problem_number in range(1, 24):
 
     if torch.cuda.is_available(): model.cuda()
 
-    model_filename = model.name + '_' + \
-                     str(problem_number) + '_' + \
+    model_filename = model.name + '_pb:' + \
+                     str(problem_number) + '_ns:' + \
                      int_to_suffix(args.nb_train_samples) + '.param'
 
     nb_parameters = 0