Upload v1-finetune_blacklight.yaml
Browse files
v1-finetune_blacklight.yaml
CHANGED
@@ -102,7 +102,7 @@ lightning:
|
|
102 |
|
103 |
trainer:
|
104 |
benchmark: True
|
105 |
-
max_epochs:
|
106 |
max_steps: 99000 # better to end on epochs not steps, especially with >500 images to ensure even distribution, but you can set this if you really want...
|
107 |
check_val_every_n_epoch: 1
|
108 |
gpus: 0,
|
|
|
102 |
|
103 |
trainer:
|
104 |
benchmark: True
|
105 |
+
max_epochs: 1 # suggest 3-4+ and adjust repeats above, this will give you a few ckpts to test
|
106 |
max_steps: 99000 # better to end on epochs not steps, especially with >500 images to ensure even distribution, but you can set this if you really want...
|
107 |
check_val_every_n_epoch: 1
|
108 |
gpus: 0,
|