jpterry commited on
Commit
467722d
·
1 Parent(s): ff9d9a6
Files changed (1) hide show
  1. model_utils/efficientnet_config.py +2 -2
model_utils/efficientnet_config.py CHANGED
@@ -14,8 +14,8 @@ from torch import Tensor, nn
14
  from torchvision.models._utils import _make_divisible
15
  from torchvision.ops import StochasticDepth
16
 
17
- sys.path.insert(1, "../")
18
- from vision_modifications import Conv2dNormActivation, SqueezeExcitation
19
 
20
 
21
  @dataclass
 
14
  from torchvision.models._utils import _make_divisible
15
  from torchvision.ops import StochasticDepth
16
 
17
+ sys.path.insert(1, "./")
18
+ from .vision_modifications import Conv2dNormActivation, SqueezeExcitation
19
 
20
 
21
  @dataclass