Message is printed when the model is saved.

bharatr21 · May 31, 2018 · b740d5a · b740d5a
1 parent d8a8dda
commit b740d5a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/rl/scripts/train_il.py b/rl/scripts/train_il.py
@@ -186,5 +186,6 @@
         if torch.cuda.is_available():
             acmodel.cpu()
         utils.save_model(acmodel, model_name)
+        logger.log("Model is saved.")
         if torch.cuda.is_available():
             acmodel.cuda()
diff --git a/rl/scripts/train_rl.py b/rl/scripts/train_rl.py
@@ -180,5 +180,6 @@
         if torch.cuda.is_available():
             acmodel.cpu()
         utils.save_model(acmodel, model_name)
+        log("Model is saved.")
         if torch.cuda.is_available():
             acmodel.cuda()