基于pytorch的自动下载VGG16预训练模型的四种天气分类

使用情况：
model = torchvision.models.vgg16(pretrained=True)
将训练好的卷积层参数冻结，抛弃训练好的线性层，然后使用自己搭建的好线性层。

import torch
import torch.nn as nn
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import torch.nn.functional as F
import torchvision
import time
import os
import shutil
from torchvision import datasets,transforms
from torch.utils.data import TensorDataset
from torch.utils.data import DataLoader
from sklearn.model_selection import train_test_split

start=time.time()

base_dir = r'D:/pycharmworkspace/ISLR-master/fourweather'
train_dir = os.path.join(base_dir,'train')
test_dir = os.path.join(base_dir,'test')

transformation = transforms.Compose([
    transforms.Resize((192,192)),
    transforms.ToTensor(),
    transforms.Normalize(mean=[0.5,0.5,0.5],std=