问题描述
Train acc持续上升,直至80%+;但是Valid acc只有1%~3%,最高也就20%+
原来的代码如下
# ========================= step 1/5 数据 =========================
norm_mean = [0.485, 0.456, 0.406]
norm_std = [0.229, 0.224, 0.225]
normalizes = transforms.Normalize(norm_mean, norm_std)
# 训练阶段图片预处理(数据增强)
train_transform = transforms.Compose([
transforms.Resize((128, 128)), # (256, 256) 区别
# transforms.CenterCro