pretrained=False怎么修改为torchvision0.15版本中的写法
时间: 2024-03-06 21:49:58 浏览: 56
doris 0.15版本 中的 apache_hdfs_broker 插件
在torchvision 0.15中,如果您不想使用预训练模型,则可以将`pretrained=False`修改为`weights=None`,如下所示:
```
resnet = models.resnet34(weights=None)
```
这将创建一个未经过预训练的ResNet-34模型。如果您使用其他模型,请将`resnet34`修改为相应的模型名称。
阅读全文