__pydantic_initialised__: true | |
batch_size: 5 | |
device: cpu | |
hour_embed_type: none | |
hub_private_repo: false | |
learning_rate: 0.0001 | |
lr_warmup_steps: 500 | |
mixed_precision: fp16 | |
num_epochs: 50 | |
num_samples: 3 | |
output_dir: nn-probando-tiny | |
overwrite_output_dir: true | |
push_to_hub: false | |
save_image_epochs: 2 | |
save_model_epochs: 5 | |
seed: 2023 | |