diff options
Diffstat (limited to 'training/conf/experiment/vqvae.yaml')
-rw-r--r-- | training/conf/experiment/vqvae.yaml | 51 |
1 files changed, 0 insertions, 51 deletions
diff --git a/training/conf/experiment/vqvae.yaml b/training/conf/experiment/vqvae.yaml deleted file mode 100644 index d069aef..0000000 --- a/training/conf/experiment/vqvae.yaml +++ /dev/null @@ -1,51 +0,0 @@ -defaults: - - override /network: vqvae - - override /criterion: mse - - override /model: lit_vqvae - - override /callbacks: wandb_vae - - override /optimizers: null - # - override /lr_schedulers: - # - cosine_annealing - -# lr_schedulers: null -# network: -# _target_: torch.optim.lr_scheduler.OneCycleLR -# max_lr: 1.0e-2 -# total_steps: null -# epochs: 100 -# steps_per_epoch: 200 -# pct_start: 0.1 -# anneal_strategy: cos -# cycle_momentum: true -# base_momentum: 0.85 -# max_momentum: 0.95 -# div_factor: 25 -# final_div_factor: 1.0e4 -# three_phase: true -# last_epoch: -1 -# verbose: false - -# # Non-class arguments -# interval: step -# monitor: val/loss - -optimizers: - network: - _target_: madgrad.MADGRAD - lr: 1.0e-4 - momentum: 0.9 - weight_decay: 0 - eps: 1.0e-7 - - parameters: network - -trainer: - max_epochs: 128 - limit_train_batches: 0.1 - limit_val_batches: 0.1 - -datamodule: - batch_size: 8 - # resize: [288, 320] - -summary: null |