summaryrefslogtreecommitdiff
path: root/training/conf/model/lit_vqvae.yaml
diff options
context:
space:
mode:
Diffstat (limited to 'training/conf/model/lit_vqvae.yaml')
-rw-r--r--training/conf/model/lit_vqvae.yaml2
1 files changed, 1 insertions, 1 deletions
diff --git a/training/conf/model/lit_vqvae.yaml b/training/conf/model/lit_vqvae.yaml
index 7136dbd..6be37e5 100644
--- a/training/conf/model/lit_vqvae.yaml
+++ b/training/conf/model/lit_vqvae.yaml
@@ -1,3 +1,3 @@
-type: LitVQVAEModel
+_target_: text_recognizer.models.vqvae.VQVAELitModel
args:
mapping: sentence_piece