Update custom_interface.py

custom_interface.py

@@ -18,6 +18,7 @@ class ASR(Pretrained):
         embedded_tokens = self.mods.embedding(tokens_bos)
         decoder_outputs, _ = self.mods.decoder(embedded_tokens, encoded_outputs, self.wav_lens)
 
+        print(decoder_outputs)
         # Output layer for seq2seq log-probabilities
         predictions = self.hparams.test_search(encoded_outputs, self.wav_lens)[0]
         predicted_words = [self.hparams.tokenizer.decode_ids(prediction).split(" ") for prediction in predictions]