jpterry commited on
Commit
c44b8b8
·
1 Parent(s): 960d507
Files changed (1) hide show
  1. model_utils/efficientnet_config.py +1 -0
model_utils/efficientnet_config.py CHANGED
@@ -14,6 +14,7 @@ from torch import Tensor, nn
14
  from torchvision.models._utils import _make_divisible
15
  from torchvision.ops import StochasticDepth
16
 
 
17
  from vision_modifications import Conv2dNormActivation, SqueezeExcitation
18
 
19
 
 
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