File size: 422 Bytes
27ca8b3 |
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 |
defaults:
- base_pytorch
tasks: [training]
training:
lr: 8e-5
precision: 16-mixed
batch_size: 4
max_epochs: -1
max_steps: 2000005
checkpointing:
every_n_train_steps: 2500
optim:
gradient_clip_val: 1.0
validation:
val_every_n_step: 300
val_every_n_epoch: null
batch_size: 4
limit_batch: 1
test:
limit_batch: 1
batch_size: 1
logging:
metrics:
# - fvd
# - fid
# - lpips
|