diff --git a/models.py b/models.py index 94f357b3..4c00d148 100755 --- a/models.py +++ b/models.py @@ -4,7 +4,7 @@ from pathlib import Path import torch.nn.functional as F -ONNX_EXPORT = True +ONNX_EXPORT = False def create_modules(module_defs):