batch_size: 2 device: cuda hour_embed_type: class hub_private_repo: false learning_rate: 0.0001 lr_warmup_steps: 500 mixed_precision: fp16 num_epochs: 100 num_samples: 3 number_model_params: 155533 output_dir: cddpm-tiny-bicubic overwrite_output_dir: true push_to_hub: true save_image_epochs: 10 save_model_epochs: 5 seed: 2023