我现在是resnet34模型,但我想换其他模型对比一下,请问我应该怎么做
如果你使用的是Pytorch深度学习框架,torchvision.models 里有多个模型并且有相应的预训练参数,如下可直接修改模型:自定义的模型需要自定义模型结构。
model = torchvision.models.resnet18()
```