pereza's picture
Epoch 1
50338ae
__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