File size: 342 Bytes
50338ae |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
__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: 74011
output_dir: vqvae-tiny-small
overwrite_output_dir: true
push_to_hub: true
save_image_epochs: 5
save_model_epochs: 10
seed: 2023
|