Update v1-finetune_unfrozen.yaml

SujeethJinesh · Sep 9, 2022 · add76d4 · add76d4
1 parent 00e984c
commit add76d4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/configs/stable-diffusion/v1-finetune_unfrozen.yaml b/configs/stable-diffusion/v1-finetune_unfrozen.yaml
@@ -93,7 +93,7 @@ data:
         set: train
         reg: true
         per_image_tokens: false
-        repeats: 100
+        repeats: 10
 
     validation:
       target: ldm.data.personalized.PersonalizedBase
@@ -117,4 +117,4 @@ lightning:
 
   trainer:
     benchmark: True
-    max_steps: 800
+    max_steps: 800