Update v1-finetune_runpod.yaml

This commit is contained in:
Victor Hall 2023-01-02 13:41:07 -05:00 committed by GitHub
parent 99a9bc758d
commit 1b5dff0e3a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

View File

@ -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,