update

examples/cnn_vad_by_webrtcvad/step_4_train_model.py

@@ -264,6 +264,7 @@ def main():
             num_samples = noisy_audios.shape[-1]
 
             logits, probs, lsnr = model.forward(noisy_audios)
+            lsnr = torch.squeeze(lsnr, dim=-1)
 
             targets = BaseVadLoss.get_targets(probs, batch_vad_segments, duration=num_samples / config.sample_rate)