From 08ef6b7c332153cd72b7a225e27ee7af8882f313 Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Fri, 16 Jun 2017 08:30:15 +0200 Subject: [PATCH] Ooops. --- cnn-svrt.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/cnn-svrt.py b/cnn-svrt.py index 6645ac1..7bef242 100755 --- a/cnn-svrt.py +++ b/cnn-svrt.py @@ -171,15 +171,15 @@ for arg in vars(args): for problem_number in range(1, 24): - model_filename = model.name + '_' + \ - str(problem_number) + '_' + \ - str(args.nb_train_batches) + '.param' - model = AfrozeShallowNet() if torch.cuda.is_available(): model.cuda() + model_filename = model.name + '_' + \ + str(problem_number) + '_' + \ + str(args.nb_train_batches) + '.param' + nb_parameters = 0 for p in model.parameters(): nb_parameters += p.numel() log_string('nb_parameters {:d}'.format(nb_parameters)) -- 2.20.1