export.py folder sys.path addition (#1052)

2020-10-04 17:42:05 +02:00 · 2020-10-04 17:42:05 +02:00 · d55c326dda
parent 7220cee1d1
commit d55c326dda
1 changed files with 3 additions and 0 deletions
--- 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