Skip to content

Commit 411ce25

Browse files
committed
Adding code skeleton
1 parent 759c5b6 commit 411ce25

File tree

1 file changed

+27
-0
lines changed

1 file changed

+27
-0
lines changed

torchvision/models/efficientnet.py

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -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+
pass
20+
21+
22+
class EfficientNet(nn.Module):
23+
pass
24+
25+
26+
def efficientnet_b0(pretrained: bool = False, progress: bool = True, **kwargs: Any) -> EfficientNet:
27+
pass

0 commit comments

Comments
 (0)