Update train.py

zhenye234 · Oct 28, 2023 · a9d1eaa · a9d1eaa
1 parent 255856e
commit a9d1eaa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -87,7 +87,7 @@
         model = Comospeech(nsymbols, 1, None, n_enc_channels, filter_channels, filter_channels_dp, 
                         n_heads, n_enc_layers, enc_kernel, enc_dropout, window_size, 
                         n_feats,teacher=False).cuda()
-        mode1 = load_teacher_model(model,checkpoint_dir='') # teacher model path
+        model = load_teacher_model(model,checkpoint_dir='') # teacher model path
         optimizer = torch.optim.Adam(params=model.decoder.denoise_fn.parameters(), lr=learning_rate)