jpxkqx's picture
Epoch 30
cb5d3ab
raw
history blame
362 Bytes
__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