upadte docs

dingguanglei · Apr 19, 2019 · 817521c · 817521c
1 parent 3941d6f
commit 817521c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/jdit/trainer/__init__.py b/jdit/trainer/__init__.py
@@ -3,6 +3,7 @@
 from .single import *
 # from .instances import *
 from jdit.trainer import instances
+from jdit.trainer import single
 
-__all__ = ['SupTrainer', 'ClassificationTrainer', 'instances', 'SupGanTrainer', 'Pix2pixGanTrainer',
+__all__ = ['SupTrainer', 'ClassificationTrainer', 'instances', 'single', 'SupGanTrainer', 'Pix2pixGanTrainer',
            'GenerateGanTrainer']