HoneyTian commited on
Commit
555ea1a
·
1 Parent(s): f4e5ea8
examples/sound_classification_by_lstm/step_9_evaluation_onnx_model.py CHANGED
@@ -92,6 +92,7 @@ def main():
92
  window_fn=torch.hamming_window if config.mel_spectrogram_param["window_fn"] == "hamming" else torch.hann_window,
93
  n_mels=config.mel_spectrogram_param["n_mels"],
94
  )
 
95
 
96
  df = pd.read_excel(evaluation_file.as_posix())
97
  result = list()
 
92
  window_fn=torch.hamming_window if config.mel_spectrogram_param["window_fn"] == "hamming" else torch.hann_window,
93
  n_mels=config.mel_spectrogram_param["n_mels"],
94
  )
95
+ wave_to_mel_spectrogram.to(device)
96
 
97
  df = pd.read_excel(evaluation_file.as_posix())
98
  result = list()