Spaces:
Running
on
Zero
Running
on
Zero
File size: 499 Bytes
b213d84 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
# for virtual try-on
# height: 512
# width: 384
# batch_size: 8
height: 1024
width: 768
batch_size: 2
# for pose transfer
# height: 256
# width: 176
# batch_size: 8
# height: 512
# width: 352
# batch_size: 4
# height: 1024
# width: 704
# batch_size: 1
precision: bf16
max_steps: null
max_epochs: 200
max_train_steps_per_epoch: null
evaluate_every_n_train_steps: null
evaluate_every_n_train_epochs: null
max_eval_steps_per_eval_epoch: null
use_torchsnapshot: false
checkpoint_every_n_steps: 500
|