Half training paramters for SSD

lucataco · Nov 12, 2023 · e891c2f · e891c2f
1 parent 4499292
commit e891c2f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train.py b/train.py
@@ -39,11 +39,11 @@ def train(
     ),
     num_train_epochs: int = Input(
         description="Number of epochs to loop through your training dataset",
-        default=4000,
+        default=2000,
     ),
     max_train_steps: int = Input(
         description="Number of individual training steps. Takes precedence over num_train_epochs",
-        default=1000,
+        default=500,
     ),
     # gradient_accumulation_steps: int = Input(
     #     description="Number of training steps to accumulate before a backward pass. Effective batch size = gradient_accumulation_steps * batch_size",