Mirrors
/
EveryDream-trainer
mirror of https://github.com/victorchall/EveryDream-trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update v1-finetune_micro.yaml

This commit is contained in:
Victor Hall 2023-01-02 13:40:50 -05:00 committed by GitHub
parent 7dbb348f57
commit 99a9bc758d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
configs/stable-diffusion/v1-finetune_micro.yaml
View File

@ -78,7 +78,7 @@ data:
validation: validation:
target: ldm.data.ed_validate.EDValidateBatch target: ldm.data.ed_validate.EDValidateBatch
params: params:
repeats: 10 repeats: 0.01
test: test:
target: ldm.data.ed_validate.EDValidateBatch target: ldm.data.ed_validate.EDValidateBatch
params: params:
@ -105,4 +105,4 @@ lightning:
max_epochs: 4 # suggest 3-4+ and adjust repeats above, this will give you a few ckpts to test max_epochs: 4 # suggest 3-4+ and adjust repeats above, this will give you a few ckpts to test
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... 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 check_val_every_n_epoch: 1
gpus: 0, gpus: 0,