summaryrefslogtreecommitdiff
path: root/training/conf/lr_scheduler
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-08-10 23:25:13 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-08-10 23:25:13 +0200
commit441b7484348953deb7c94150675d54583ef5a81a (patch)
tree6a3e537f98a5c34fe166db7f4c60552b1212b8f5 /training/conf/lr_scheduler
parent82f4acabe24e5171c40afa2939a4777ba87bcc30 (diff)
Update to config and logging in VQGAN
Diffstat (limited to 'training/conf/lr_scheduler')
-rw-r--r--training/conf/lr_scheduler/cosine_annealing.yaml8
-rw-r--r--training/conf/lr_scheduler/one_cycle.yaml20
2 files changed, 0 insertions, 28 deletions
diff --git a/training/conf/lr_scheduler/cosine_annealing.yaml b/training/conf/lr_scheduler/cosine_annealing.yaml
deleted file mode 100644
index c53ee3a..0000000
--- a/training/conf/lr_scheduler/cosine_annealing.yaml
+++ /dev/null
@@ -1,8 +0,0 @@
-cosine_annealing:
- _target_: torch.optim.lr_scheduler.CosineAnnealingLR
- T_max: 256
- eta_min: 0.0
- last_epoch: -1
-
- interval: epoch
- monitor: val/loss
diff --git a/training/conf/lr_scheduler/one_cycle.yaml b/training/conf/lr_scheduler/one_cycle.yaml
deleted file mode 100644
index c60577a..0000000
--- a/training/conf/lr_scheduler/one_cycle.yaml
+++ /dev/null
@@ -1,20 +0,0 @@
-onc_cycle:
- _target_: torch.optim.lr_scheduler.OneCycleLR
- max_lr: 1.0e-3
- total_steps: null
- epochs: 512
- steps_per_epoch: 4992
- pct_start: 0.3
- anneal_strategy: cos
- cycle_momentum: true
- base_momentum: 0.85
- max_momentum: 0.95
- div_factor: 25.0
- final_div_factor: 10000.0
- three_phase: true
- last_epoch: -1
- verbose: false
-
- # Non-class arguments
- interval: step
- monitor: val/loss