__pydantic_initialised__: true | |
device: cuda | |
hour_embed_type: class | |
hub_private_repo: false | |
learning_rate: 0.0001 | |
lr_warmup_steps: 500 | |
mixed_precision: fp16 | |
num_epochs: 200 | |
num_samples: 3 | |
number_model_params: 57579 | |
output_dir: vqvae-tiny-small | |
overwrite_output_dir: true | |
push_to_hub: true | |
save_image_epochs: 5 | |
save_model_epochs: 10 | |
seed: 2023 | |