| defaults: | |
| - base_pytorch | |
| tasks: [training] | |
| training: | |
| lr: 2e-5 | |
| precision: 16-mixed | |
| batch_size: 4 | |
| # batch_size: 8 | |
| max_epochs: -1 | |
| max_steps: 2000005 | |
| checkpointing: | |
| every_n_train_steps: 2500 | |
| optim: | |
| gradient_clip_val: 1.0 | |
| validation: | |
| val_every_n_step: 2500 | |
| val_every_n_epoch: null | |
| batch_size: 4 | |
| limit_batch: 1 | |
| test: | |
| limit_batch: 1 | |
| batch_size: 1 | |
| logging: | |
| metrics: | |
| # - fvd | |
| # - fid | |
| # - lpips | |