torchvision官网的分类模型 <notitle>—Torchvision0.20documentation训练和预测时，改成自己的分类数#model=models.resnet50(weights=models.ResNet50_Weights.IMAGENET1K_V2)#加载模型。会自动下载模型#model=models.resnet18(weights=models.ResNet18_