python 深度学习 训练流程梳理--pytorch简单搭建自己的训练过程
1.数据集的准备(官方Data–CIFAR10为例),主要包括两部分:train_data、test_data:
import 需要的包,小技巧:开始不知道用哪些包可以先不引用,输入命令后未引用的包下面会有红线警告,鼠标点击一下,然后 alt+enter会显示 需要引用的选项,选择即可。
train_data = torchvision.datasets.CIFAR10(root= r'D:\train_maskcsv\Tudui\data',train=True,transform=torchvisio
原创
2021-12-20 16:20:57 ·
2025 阅读 ·
1 评论