Spaces:
Sleeping
Sleeping
update
Browse files
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()
|