__pydantic_initialised__: true | |
device: cpu | |
hour_embed_type: none | |
hub_private_repo: false | |
learning_rate: 0.0001 | |
lr_warmup_steps: 500 | |
mixed_precision: fp16 | |
num_epochs: 100 | |
num_samples: 3 | |
output_dir: cddpm-probando-tiny | |
overwrite_output_dir: true | |
push_to_hub: true | |
save_image_epochs: 10 | |
save_model_epochs: 30 | |
seed: 2023 | |
train_batch_size: 32 | |
val_batch_size: 4 | |