From 8a7266130538c9c7c7f02b2654d8569630447f55 Mon Sep 17 00:00:00 2001 From: Vahid Behzadan Date: Tue, 23 Jan 2018 17:56:09 -0600 Subject: [PATCH] revised train.py - saves all checkpoints --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index ebbe4bd..1cf482b 100644 --- a/train.py +++ b/train.py @@ -80,7 +80,7 @@ def maybe_save_model(savedir, container, state): U.save_state(os.path.join(savedir, model_dir, "saved")) if container is not None: container.put(os.path.join(savedir, model_dir), model_dir) - #relatively_safe_pickle_dump(state, os.path.join(savedir, 'training_state.pkl.zip'), compression=True) + relatively_safe_pickle_dump(state, os.path.join(savedir, 'training_state.pkl.zip'), compression=True) if container is not None: container.put(os.path.join(savedir, 'training_state.pkl.zip'), 'training_state.pkl.zip') relatively_safe_pickle_dump(state["monitor_state"], os.path.join(savedir, 'monitor_state.pkl'))