add training configuration in cogvideox

a-r-r-o-w · Jan 13, 2025 · cfbb5da · cfbb5da
1 parent 9f2d5d0
commit cfbb5da
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/docs/training/cogvideox.md b/docs/training/cogvideox.md
@@ -115,6 +115,19 @@ Training configuration: {
 
 ### Full finetuning
 
+```
+Training configuration: {
+    "trainable parameters": 5570283072,
+    "total samples": 1,
+    "train epochs": 2,
+    "train steps": 2,
+    "batches per device": 1,
+    "total batches observed per epoch": 1,
+    "train batch size": 1,
+    "gradient accumulation steps": 1
+}
+```
+
 | stage                         | memory_allocated  | max_memory_reserved |
 |:-----------------------------:|:-----------------:|:-------------------:|
 | after precomputing conditions |  8.880            | 8.941               |