diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp index 453360e32..18f15f79d 100644 Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode.py index c6fc9981e..67eb7f5df 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode.py @@ -95,9 +95,14 @@ class LoRAHook(): #print(input[0].size(), output.size()) #print('-'*20) + print(output) + lora_out = self.lora(input[0]) output += lora_out + print(output) + print('*'*20) + def save_checkpoint(self, i, iter_, save_dir): if isinstance(self.lora, DDP): lora = self.lora.module