jpxkqx's picture
Epoch 30
6b2b8a6
raw
history blame
356 Bytes
__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: 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