X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pysvrt.git;a=blobdiff_plain;f=cnn-svrt.py;h=cfef09da934fb5d82181715f428dab1a261d04ba;hp=d44de2aed08f7097f6c255587d035ede105a14bd;hb=37030c396217cfe89f8dfa2b9e10ff1ec783a5a7;hpb=a0de65c26758ec247ff419b9b53725733ed0e76c 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