From b44de0e11281c723ec426f8bec8ca0897ecfe3ff Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sun, 21 Nov 2021 21:34:53 +0100 Subject: Remove VQVAE stuff, did not work... --- training/conf/experiment/vqvae.yaml | 51 ------------------------------------- 1 file changed, 51 deletions(-) delete mode 100644 training/conf/experiment/vqvae.yaml (limited to 'training/conf/experiment/vqvae.yaml') 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 -- cgit v1.2.3-70-g09d2