Allow a new maxepoch when resuming from a checkpoint.

facebookresearch · Jun 6, 2017 · 1def963 · 1def963
1 parent 501d29c
commit 1def963
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/train.lua b/train.lua
@@ -396,6 +396,12 @@ if plpath.isfile(lastStatePath) and not config.nosave then
                 state.epoch, state.t, state.epoch_t, state.jumped))
         end
     end
+
+    -- Support modifying the maxepoch setting during resume
+    engine.hooks.onResume = function(state)
+        state.maxepoch = config.maxepoch
+    end
+
     engine:resume{
         path = lastStatePath,
         iterator = corpus.train,