Update extract_feature_print.py
Browse files- extract_feature_print.py +2 -2
extract_feature_print.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
import os, sys, traceback
|
| 2 |
-
|
| 3 |
|
| 4 |
os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
|
| 5 |
os.environ["PYTORCH_MPS_HIGH_WATERMARK_RATIO"] = "0.0"
|
|
@@ -37,7 +37,7 @@ def printt(strr):
|
|
| 37 |
|
| 38 |
|
| 39 |
printt(sys.argv)
|
| 40 |
-
model_path = feature_extraction_method
|
| 41 |
|
| 42 |
printt(exp_dir)
|
| 43 |
wavPath = "%s/1_16k_wavs" % exp_dir
|
|
|
|
| 1 |
import os, sys, traceback
|
| 2 |
+
|
| 3 |
|
| 4 |
os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
|
| 5 |
os.environ["PYTORCH_MPS_HIGH_WATERMARK_RATIO"] = "0.0"
|
|
|
|
| 37 |
|
| 38 |
|
| 39 |
printt(sys.argv)
|
| 40 |
+
model_path = globals().get("feature_extraction_method")
|
| 41 |
|
| 42 |
printt(exp_dir)
|
| 43 |
wavPath = "%s/1_16k_wavs" % exp_dir
|