__pydantic_initialised__: true | |
batch_size: 16 | |
device: cuda | |
hour_embed_type: none | |
hub_private_repo: false | |
learning_rate: 0.0001 | |
lr_warmup_steps: 100 | |
mixed_precision: fp16 | |
num_epochs: 100 | |
num_samples: 3 | |
output_dir: nn-tiny-subset-domain | |
overwrite_output_dir: true | |
push_to_hub: false | |
save_image_epochs: 5 | |
save_model_epochs: 5 | |
seed: 2023 | |