From 4da7a2c812221d56a430b35139ac40b23fa76f77 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Tue, 29 Jun 2021 22:54:52 +0200
Subject: Refactor of config, more granular

---
 training/conf/model/lit_vqvae.yaml | 23 +----------------------
 1 file changed, 1 insertion(+), 22 deletions(-)

(limited to 'training/conf/model')

diff --git a/training/conf/model/lit_vqvae.yaml b/training/conf/model/lit_vqvae.yaml
index 90780b7..7136dbd 100644
--- a/training/conf/model/lit_vqvae.yaml
+++ b/training/conf/model/lit_vqvae.yaml
@@ -1,24 +1,3 @@
-# @package _group_
 type: LitVQVAEModel
 args:
-    optimizer:
-        type: MADGRAD
-        args:
-            lr: 1.0e-3
-            momentum: 0.9
-            weight_decay: 0
-            eps: 1.0e-6
-    lr_scheduler:
-        type: OneCycleLR
-        args:
-            interval: step
-            max_lr: 1.0e-3
-            three_phase: true
-            epochs: 64
-            steps_per_epoch: 633 # num_samples / batch_size
-    criterion:
-        type: MSELoss
-        args:
-            reduction: mean
-    monitor: val_loss
-    mapping: sentence_piece
+  mapping: sentence_piece
-- 
cgit v1.2.3-70-g09d2