setup 0.0.10v.

fix checkpoint
dingguanglei · Mar 22, 2019 · 897764a · 897764a
1 parent 3b9eacd
commit 897764a
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 5 deletions.
diff --git a/jdit/model.py b/jdit/model.py
@@ -250,7 +250,7 @@ def check_point(self, model_name: str, epoch: int, logdir="log"):
     def is_checkpoint(self, model_name: str, epoch: int, logdir="log"):
         if not self.check_point_pos:
             return False
-        if isinstance(epoch, int):
+        if isinstance(self.check_point_pos, int):
             is_check_point = epoch > 0 and (epoch % self.check_point_pos) == 0
         else:
             is_check_point = epoch in self.check_point_pos

diff --git a/jdit/trainer/instances/fashingClassification.py b/jdit/trainer/instances/fashingClassification.py
@@ -79,6 +79,7 @@ def start_fashingClassTrainer(gpus=(), nepochs=10, run_type="train"):
 
     print('===> Build dataset')
     mnist = FashionMNIST(batch_size=batch_size)
+    # mnist.dataset_train = mnist.dataset_test
     torch.backends.cudnn.benchmark = True
     print('===> Building model')
     net = Model(SimpleModel(depth=depth), gpu_ids_abs=gpus, init_method="kaiming", check_point_pos=1)

diff --git a/jdit/trainer/super.py b/jdit/trainer/super.py
@@ -84,10 +84,11 @@ def __setattr__(self, key, value):
             if is_change:
                 super(SupTrainer, self).__getattribute__("_record_configs")("optimizer")
         elif key == "current_epoch" and value != 0:
-            is_change = super(SupTrainer, self).__getattribute__("_change_lr")("epoch", value)
-            if is_change:
+            is_change_lr = super(SupTrainer, self).__getattribute__("_change_lr")("epoch", value)
+            if is_change_lr:
                 super(SupTrainer, self).__getattribute__("_record_configs")("optimizer")
-                super(SupTrainer, self).__getattribute__("_check_point")()
+            super(SupTrainer, self).__getattribute__("_check_point")()
+
             super(SupTrainer, self).__getattribute__("_record_configs")("performance")
         elif isinstance(value, Model):
             super(SupTrainer, self).__getattribute__("_models").update({key: value})

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name="jdit",  # pypi中的名称，pip或者easy_install安装时使用的名称，或生成egg文件的名称
-        version="0.0.9",
+        version="0.0.10",
     author="Guanglei Ding",
     author_email="dingguanglei.bupt@qq.com",
     maintainer='Guanglei Ding',