4_使用预训练模型 微调训练CIFAR10

使用预训练模型 微调训练CIFAR10

1. VGG

• 准备工作

import torch
from torch import nn
import torchvision
from torchvision import models
from torchvision import datasets, transforms
from datetime import datetime
from tqdm import tqdm
from torchsummary import summary

# dataset
num_classes = 
1
2
3
4
5
6
7
8
9
10