X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pysvrt.git;a=blobdiff_plain;f=cnn-svrt.py;h=338e145c52197e5f4649f031a83a88870473de12;hp=7dc6dfff1d6ee98df36ab92624c4ad901f3acc29;hb=141ec1e0e68af3241cfe6a28c00978e85fd5a747;hpb=9ee9a775ccd2391990b3ab226e73c86bd19bd36a diff --git a/cnn-svrt.py b/cnn-svrt.py index 7dc6dff..338e145 100755 --- a/cnn-svrt.py +++ b/cnn-svrt.py @@ -540,7 +540,10 @@ for problem_number in map(int, args.problems.split(',')): else: validation_set = None - train_model(model, model_filename, train_set, validation_set, nb_epochs_done = nb_epochs_done) + train_model(model, model_filename, + train_set, validation_set, + nb_epochs_done = nb_epochs_done) + log_string('saved_model ' + model_filename) nb_train_errors = nb_errors(model, train_set)