目录
1.定义模块
自定义Movilenet_v2类
class Mobilenet_v2(nn.Module): def __init__(self): super().__init__() model = models.mobilenet_v2(pretrained=True) self.layer=nn.Sequential( model.features, ) def forward(self, x): x=self.layer(x) x = nn.functional.adaptive_avg_pool2d(x, (1, 1)) return x