update

examples/sound_classification_by_lstm/run_batch_sound.sh

@@ -11,11 +11,11 @@
 
 # sound by language ch64
 
-sh run.sh --stage 0 --stop_stage 7 --system_version centos --file_folder_name file_dir --final_model_name sound-en-ph-3-l3-ch64-lstm \
---filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/en-PH/wav_finished/*/*.wav" \
---label_plan 3 \
---config_file "yaml/lstm-classifier-3-l3-ch64.yaml" \
---pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/sound-3-l3-ch64-lstm.zip"
+#sh run.sh --stage 0 --stop_stage 7 --system_version centos --file_folder_name file_dir --final_model_name sound-en-ph-3-l3-ch64-lstm \
+#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/en-PH/wav_finished/*/*.wav" \
+#--label_plan 3 \
+#--config_file "yaml/lstm-classifier-3-l3-ch64.yaml" \
+#--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/sound-3-l3-ch64-lstm.zip"
 
 sh run.sh --stage 0 --stop_stage 7 --system_version centos --file_folder_name file_dir --final_model_name sound-en-sg-3-l3-ch64-lstm \
 --filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/en-SG/wav_finished/*/*.wav" \

examples/sound_classification_by_lstm/step_9_evaluation_onnx_model.py

@@ -42,7 +42,7 @@ def get_args():
     parser.add_argument("--model_dir", default="best", type=str)
     parser.add_argument("--onnx_model_file", default="model.onnx", type=str)
     parser.add_argument("--output_file", default="evaluation_onnx.xlsx", type=str)
-    parser.add_argument("--device", default="cpu", type=str)
+    parser.add_argument("--device", default="cuda" if torch.cuda.is_available() else "cpu", type=str)
 
     parser.add_argument("--max_count", default=10000, type=int)