pytorch 迁移学习 猫狗大战 note
import osfrom pathlib import Pathfrom PIL import Imagefrom torch.utils.data import DataLoader,Datasetimport numpy as npfrom torchvision import transforms, modelsimport matplotlib.pyplot as pltfrom sklearn.model_selection import train_test_splitimpo