微調(fine-tuning)是指利用已經訓練好的模型,在新的數據集上進行重新訓練的過程,以適應新的數據集。在PyTorch中,可以通過以下步驟實現微調:
import torch
import torchvision.models as models
model = models.resnet18(pretrained=True)
num_features = model.fc.in_features
model.fc = torch.nn.Linear(num_features, num_classes) # num_classes為新的數據集的類別數
criterion = torch.nn.CrossEntropyLoss()
optimizer = torch.optim.SGD(model.parameters(), lr=0.001, momentum=0.9)
# 假設 dataloaders 包含訓練和驗證的 DataLoader
for epoch in range(num_epochs):
model.train()
for inputs, labels in dataloaders['train']:
optimizer.zero_grad()
outputs = model(inputs)
loss = criterion(outputs, labels)
loss.backward()
optimizer.step()
# 在驗證集上評估模型性能
model.eval()
with torch.no_grad():
for inputs, labels in dataloaders['val']:
outputs = model(inputs)
# 計算準確率等指標
通過以上步驟,即可完成在PyTorch中進行模型微調的過程。需要注意的是,在微調過程中,可以根據具體情況調整學習率、訓練策略等參數以達到更好的效果。