File size: 362 Bytes
cb5d3ab |
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: positional
hub_private_repo: false
learning_rate: 0.0001
lr_warmup_steps: 500
mixed_precision: fp16
num_epochs: 100
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
|