Made the model filenames more explicit.

[pysvrt.git] / cnn-svrt.py

diff --git a/cnn-svrt.py b/cnn-svrt.py
index 153bdc9..cfef09d 100755 (executable)
--- a/cnn-svrt.py
+++ b/cnn-svrt.py
@@ -255,9 +255,9 @@ for arg in vars(args):
 ######################################################################
 
 def int_to_suffix(n):
-    if n > 1000000 and n%1000000 == 0:
+    if n >= 1000000 and n%1000000 == 0:
         return str(n//1000000) + 'M'
-    elif n > 1000 and n%1000 == 0:
+    elif n >= 1000 and n%1000 == 0:
         return str(n//1000) + 'K'
     else:
         return str(n)
@@ -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