Docs update and pin keras

deepfakes · Jun 7, 2024 · 4e353df · 4e353df
1 parent 3e5966d
commit 4e353df
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 1 deletion.
diff --git a/docs/full/plugins/train.rst b/docs/full/plugins/train.rst
@@ -57,10 +57,37 @@ model.original module
 trainer package
 ===============
 
+This package contains the training loop for Faceswap
+
+.. rubric:: Module Summary
+
+.. autosummary::
+   :nosignatures:
+
+   ~plugins.train.trainer._base
+   ~plugins.train.model._display
+   ~plugins.train.model._tensorboard
+
 trainer._base module
 ----------------------
 
 .. automodule:: plugins.train.trainer._base
    :members:
    :undoc-members:
    :show-inheritance:
+
+trainer._display module
+-----------------------
+
+.. automodule:: plugins.train.trainer._display
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+trainer._tensorboard module
+---------------------------
+
+.. automodule:: plugins.train.trainer._tensorboard
+   :members:
+   :undoc-members:
+   :show-inheritance:
diff --git a/requirements/_requirements_base.txt b/requirements/_requirements_base.txt
@@ -14,4 +14,4 @@ pywin32>=305 ; sys_platform == "win32"
 torchvision>=0.17.2,<0.19.0
 torchaudio>=2.2.2,<2.4.0
 tensorboard>=2.12.1
-keras>=3.3.3
+keras==3.3.3