diff --git a/models/export.py b/models/export.py index 4715596d7..09af4677e 100644 --- a/models/export.py +++ b/models/export.py @@ -9,6 +9,9 @@ import argparse import torch import torch.nn as nn +import sys +sys.path.append('./') + import models from models.experimental import attempt_load from utils.activations import Hardswish