|
@@ -23,6 +23,7 @@ from pathlib import Path
|
|
|
import sys
|
|
import sys
|
|
|
os.environ["USE_NNPACK"] = "0"
|
|
os.environ["USE_NNPACK"] = "0"
|
|
|
os.environ["NNPACK_DISABLE"] = "1"
|
|
os.environ["NNPACK_DISABLE"] = "1"
|
|
|
|
|
+os.environ["TORCH_CPP_LOG_LEVEL"] = "ERROR"
|
|
|
import torch
|
|
import torch
|
|
|
torch.backends.nnpack.enabled = False
|
|
torch.backends.nnpack.enabled = False
|
|
|
import warnings
|
|
import warnings
|
