setup 0.0.10v.

fix checkpoint
dingguanglei · Mar 22, 2019 · d223424 · d223424
1 parent 897764a
commit d223424
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/jdit/trainer/super.py b/jdit/trainer/super.py
@@ -156,9 +156,9 @@ def debug(self):
                 item.dataset_test, _ = random_split(item.dataset_test, [2, len(item.dataset_test) - 2])
                 item.build_loaders()
             if isinstance(item, Model):
-                item.check_point_pos = 1
+                item.check_point_pos = 2
             if isinstance(item, Optimizer):
-                item.check_point_pos = 1
+                item.check_point_pos = 2
                 item.position_type = "step"
         # the tested functions
         debug_fcs = [self._record_configs, self.train_epoch, self.valid_epoch,