基本的 PyTorch 代码框架如下:
import torch
import torchvision
from efficientnet_pytorch import EfficientNet
# 定义模型
model = EfficientNet.from_name('efficientnet-b2')
# 定义损失函数和优化器
criterion = torch.nn.CrossEntropyLoss()
optimizer = torch.optim.Adam(model.parameters(), lr=0.001)
# 训练模型
for epoch in range(