upadte watcher

jdit/trainer/super.py

@@ -160,6 +160,7 @@ def debug(self):
                 item.dataset_valid, _ = random_split(item.dataset_valid, [2, len(item.dataset_valid) - 2])
                 item.dataset_test, _ = random_split(item.dataset_test, [2, len(item.dataset_test) - 2])
                 item.build_loaders()
+                item.sample_dataset_size = 1
                 print("datas range: (%s, %s)" % (item.samples_train[0].min().cpu().numpy(),
                                                  item.samples_train[0].max().cpu().numpy()))
             if isinstance(item, Model):
@@ -509,7 +510,7 @@ class Watcher(object):
 
     def __init__(self, logdir: str):
         self.logdir = logdir
-        self.writer = SummaryWriter(log_dir=logdir)
+        self.writer = SummaryWriter(logdir)
         self._build_dir(logdir)
         self.training_progress_images = []
         self.gif_duration = 0.5

setup.py

@@ -2,7 +2,7 @@
 
 setup(
     name="jdit",  # pypi中的名称，pip或者easy_install安装时使用的名称，或生成egg文件的名称
-    version="0.0.11",
+    version="0.0.13",
     author="Guanglei Ding",
     author_email="dingguanglei.bupt@qq.com",
     maintainer='Guanglei Ding',
@@ -26,7 +26,7 @@
     # 需要安装的依赖
     install_requires=[
         "nvidia_ml_py3>=7.352.0",
-        'imageio'
+        'imageio',
     ],
 
     # # 添加这个选项，在windows下Python目录的scripts下生成exe文件