From d3afa310f77f47553586eeee58e3d3345a754e2c Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Wed, 4 Aug 2021 05:03:51 +0200
Subject: New VQVAE

---
 training/conf/model/lit_vqvae.yaml | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'training/conf/model/lit_vqvae.yaml')

diff --git a/training/conf/model/lit_vqvae.yaml b/training/conf/model/lit_vqvae.yaml
index b337fe6..8837573 100644
--- a/training/conf/model/lit_vqvae.yaml
+++ b/training/conf/model/lit_vqvae.yaml
@@ -1,2 +1,4 @@
 _target_: text_recognizer.models.vqvae.VQVAELitModel
-mapping: sentence_piece
+interval: step
+monitor: val/loss
+latent_loss_weight: 0.25
-- 
cgit v1.2.3-70-g09d2