File size: 354 Bytes
5fbe3af |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
__pydantic_initialised__: true
device: cpu
hour_embed_type: none
hub_private_repo: false
learning_rate: 0.0001
lr_warmup_steps: 500
mixed_precision: fp16
num_epochs: 1
num_samples: 3
output_dir: cddpm-probando-tiny
overwrite_output_dir: true
push_to_hub: true
save_image_epochs: 10
save_model_epochs: 30
seed: 2023
train_batch_size: 32
val_batch_size: 4
|