Rename vq loss to commitment loss

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: