以下是一个简单的代码示例(基于PyTorch):
python
import torch
from torchvision import transforms, models
from PIL import Image
import torch.nn as nn
import torch.optim as optim
加载预训练的VGG19模型
vgg = models.vgg19(pretrained=True).features
for param in vgg.parameters():
param.requires_grad_(False)