From 37030c396217cfe89f8dfa2b9e10ff1ec783a5a7 Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Sat, 17 Jun 2017 20:59:24 +0200 Subject: [PATCH] Made the model filenames more explicit. --- cnn-svrt.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cnn-svrt.py b/cnn-svrt.py index d44de2a..cfef09d 100755 --- 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 -- 2.20.1