| batch_size: 2 | |
| device: cuda | |
| hour_embed_type: class | |
| hub_private_repo: false | |
| learning_rate: 0.0001 | |
| lr_warmup_steps: 500 | |
| mixed_precision: fp16 | |
| num_epochs: 100 | |
| num_samples: 3 | |
| number_model_params: 155533 | |
| output_dir: cddpm-tiny-bicubic | |
| overwrite_output_dir: true | |
| push_to_hub: true | |
| save_image_epochs: 10 | |
| save_model_epochs: 5 | |
| seed: 2023 | |