Spaces:
Build error
Build error
File size: 561 Bytes
daf0288 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
defaults:
- _self_
- train/lr_scheduler: exponential
- train/optimizer: adam
mode: train
img_size: [256,256]
label_type: image
max_seq_len: null
train:
epochs: 20
grad_clip: 0.2
starting_temp: 1.
temp_min: 0.06
temp_anneal_rate: 1e-6
save_every: 3
dataloader:
_target_: src.datamodule.dataloader.dataloader_vae
batch_size: 48
valid:
dataloader:
_target_: src.datamodule.dataloader.dataloader_vae
batch_size: 48
test:
metrics: null
trainer:
_target_: src.trainer.VqvaeTrainer
snapshot: null
model_weights: null
|