bug fix
Browse files
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
|