upadte docs

dingguanglei · Apr 19, 2019 · 3941d6f · 3941d6f
1 parent 2de27cc
commit 3941d6f
Showing 1 changed file with 19 additions and 1 deletion.
diff --git a/docs/source/trainer.rst b/docs/source/trainer.rst
@@ -13,12 +13,30 @@ SupTrainer
 .. autoclass:: SupTrainer
     :members:
 
+single
+------
+
+.. automodule:: jdit.trainer.single
+.. currentmodule:: jdit.trainer.single
+
+SupSingleModelTrainer
+>>>>>>>>>>>>>>>>>>>>>
+
+.. autoclass:: SupSingleModelTrainer
+    :members:
+
 ClassificationTrainer
----------------------
+>>>>>>>>>>>>>>>>>>>>>
 
 .. autoclass:: ClassificationTrainer
     :members:
 
+AutoEncoderTrainer
+>>>>>>>>>>>>>>>>>>>>>
+
+.. autoclass:: AutoEncoderTrainer
+    :members:
+
 Generative Adversarial Networks Trainer
 ---------------------------------------