ResNet50需要改进版的版本

网络备份版本，怕自己忘了

## A。这个是基础的resnet50模型样本，后期需要改进
##### 需要改进的点有
1. ~~增加时间记录time（找一下其他的notebooks~~
2. 访问自己的数据集（使用ImageFolder）
3. 封装train和test函数（自己测试另一个版本）
4. 使用冻结训练层，提高准确度
5. 用不同的优化器和损失函数
- 使用可以让结果可视化的tensorboard或Xpytorch（极客里面和抖音工地防护检测）
7. 使用不同的分类结果，比如说换成二分类
8. ~~怎么查看cifra-10数据集~~

import time

start_time = time.time()

# 运行你的代码

# 加载CIFAR10数据集
trainset = torchvision.datasets.CIFAR10(root='./data', train=True,
                                        download=True, transform=None)

# 显示第三张图片
image, label = trainset[2]
plt.imshow(image)
plt.show()


end_time = time.time()

print("程序运行时间：", end_time - start_time, "秒")

import torch
import torchvision
import torchvision.transforms as transforms
import torch.nn as nn
import torch.n