We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 759c5b6 commit 411ce25Copy full SHA for 411ce25
torchvision/models/efficientnet.py
@@ -0,0 +1,27 @@
1
+import torch
2
+
3
+from torch import nn, Tensor
4
+from torch.nn import functional as F
5
+from typing import Any
6
7
+from .._internally_replaced_utils import load_state_dict_from_url
8
9
+# TODO: refactor this to a common place?
10
+from torchvision.models.mobilenetv2 import ConvBNActivation
11
+from torchvision.models.mobilenetv3 import SqueezeExcitation
12
13
14
+class MBConvConfig:
15
+ pass
16
17
18
+class MBConv(nn.Module):
19
20
21
22
+class EfficientNet(nn.Module):
23
24
25
26
+def efficientnet_b0(pretrained: bool = False, progress: bool = True, **kwargs: Any) -> EfficientNet:
27
0 commit comments