Update v1-finetune_runpod.yaml
This commit is contained in:
parent
99a9bc758d
commit
1b5dff0e3a
|
@ -77,7 +77,7 @@ data:
|
|||
validation:
|
||||
target: ldm.data.ed_validate.EDValidateBatch
|
||||
params:
|
||||
repeats: 10
|
||||
repeats: 0.01
|
||||
test:
|
||||
target: ldm.data.ed_validate.EDValidateBatch
|
||||
params:
|
||||
|
@ -105,4 +105,4 @@ lightning:
|
|||
max_epochs: 4 # suggest 3-4+ and adjust repeats above, only "save_top_k" number (above) of epochs are kept, be mindful of volume store use
|
||||
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...
|
||||
check_val_every_n_epoch: 1
|
||||
gpus: 0,
|
||||
gpus: 0,
|
||||
|
|
Loading…
Reference in New Issue