setup0.0.11

fix docs
dingguanglei · Apr 19, 2019 · 2df8924 · 2df8924
1 parent 817521c
commit 2df8924
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 10 deletions.
diff --git a/docs/source/trainer.rst b/docs/source/trainer.rst
@@ -13,11 +13,11 @@ SupTrainer
 .. autoclass:: SupTrainer
     :members:
 
-single
-------
+Single Model Trainer
+--------------------
 
-.. automodule:: jdit.trainer.single
-.. currentmodule:: jdit.trainer.single
+.. automodule:: jdit.trainer
+.. currentmodule:: jdit.trainer
 
 SupSingleModelTrainer
 >>>>>>>>>>>>>>>>>>>>>

diff --git a/jdit/trainer/__init__.py b/jdit/trainer/__init__.py
@@ -3,7 +3,15 @@
 from .single import *
 # from .instances import *
 from jdit.trainer import instances
-from jdit.trainer import single
-
-__all__ = ['SupTrainer', 'ClassificationTrainer', 'instances', 'single', 'SupGanTrainer', 'Pix2pixGanTrainer',
-           'GenerateGanTrainer']
+# from jdit.trainer import single
+#
+# __all__ = ['SupTrainer',
+#            'ClassificationTrainer',
+#            'instances',
+#            'single',
+#            'SupGanTrainer',
+#            'Pix2pixGanTrainer',
+#            'GenerateGanTrainer',
+#            'SupSingleModelTrainer',
+#            'ClassificationTrainer',
+#            'AutoEncoderTrainer']
diff --git a/setup.py b/setup.py
@@ -15,8 +15,12 @@
     keywords="pytorch research framework",
     platforms=["all"],
     url="https://github.com/dingguanglei/jdit",
-    packages=['jdit','jdit/trainer','jdit/trainer/gan','jdit/trainer/instances',
-              'jdit/assessment','jdit/parallel'],
+    packages=['jdit','jdit/trainer',
+              'jdit/trainer/gan',
+              'jdit/trainer/single',
+              'jdit/trainer/instances',
+              'jdit/assessment',
+              'jdit/parallel'],
               # 'mypackage','mypackage/model','mypackage/metric','mypackage/model/shared'],  # 需要打包的目录列表
 
     # 需要安装的依赖