diff --git a/models.py b/models.py index 8976d519..ee0559be 100755 --- a/models.py +++ b/models.py @@ -5,7 +5,7 @@ import torch.nn.functional as F from utils.parse_config import * from utils.utils import * -ONNX_EXPORT = True +ONNX_EXPORT = False def create_modules(module_defs):