def get_network(): model = resnet18(pretrained=False) model.avgpool = nn.AdaptiveAvgPool2d(1) model.fc = nn.Linear(512, 10) return model ...
確定! 回上一頁