summaryrefslogtreecommitdiff
path: root/training/conf
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-09-11 15:44:14 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-09-11 15:44:14 +0200
commit9c829df67f0a874b2803769dc8ff3681a3c095b1 (patch)
tree974cff555d655a43f2a98830d6848adc89ead6f1 /training/conf
parent276c24bdc4817f2817b47b7a3a6bcfd9bb47b2ef (diff)
Rename vq loss to commitment loss
Diffstat (limited to 'training/conf')
-rw-r--r--training/conf/experiment/vqgan.yaml60
1 files changed, 31 insertions, 29 deletions
diff --git a/training/conf/experiment/vqgan.yaml b/training/conf/experiment/vqgan.yaml
index 485e963..9224bc7 100644
--- a/training/conf/experiment/vqgan.yaml
+++ b/training/conf/experiment/vqgan.yaml
@@ -26,36 +26,38 @@ criterion:
datamodule:
batch_size: 6
-lr_schedulers:
- generator:
- _target_: torch.optim.lr_scheduler.OneCycleLR
- max_lr: 3.0e-4
- total_steps: null
- epochs: 100
- steps_per_epoch: 3369
- pct_start: 0.1
- anneal_strategy: cos
- cycle_momentum: true
- base_momentum: 0.85
- max_momentum: 0.95
- div_factor: 1.0e3
- final_div_factor: 1.0e4
- three_phase: true
- last_epoch: -1
- verbose: false
-
- # Non-class arguments
- interval: step
- monitor: val/loss
-
- discriminator:
- _target_: torch.optim.lr_scheduler.CosineAnnealingLR
- T_max: 64
- eta_min: 0.0
- last_epoch: -1
+lr_schedulers: null
- interval: epoch
- monitor: val/loss
+# lr_schedulers:
+# generator:
+# _target_: torch.optim.lr_scheduler.OneCycleLR
+# max_lr: 3.0e-4
+# total_steps: null
+# epochs: 100
+# steps_per_epoch: 3369
+# pct_start: 0.1
+# anneal_strategy: cos
+# cycle_momentum: true
+# base_momentum: 0.85
+# max_momentum: 0.95
+# div_factor: 1.0e3
+# final_div_factor: 1.0e4
+# three_phase: true
+# last_epoch: -1
+# verbose: false
+#
+# # Non-class arguments
+# interval: step
+# monitor: val/loss
+#
+# discriminator:
+# _target_: torch.optim.lr_scheduler.CosineAnnealingLR
+# T_max: 64
+# eta_min: 0.0
+# last_epoch: -1
+#
+# interval: epoch
+# monitor: val/loss
optimizers:
generator: