![AIGC原理与实践:零基础学大语言模型、扩散模型和多模态模型](https://wfqqreader-1252317822.image.myqcloud.com/cover/190/51549190/b_51549190.jpg)
上QQ阅读APP看本书,新人免费读10天
设备和账号都新为新人
2.2.1 准备数据
1)导入必要的模块。
![](https://epubservercos.yuewen.com/B4645E/30403237107613906/epubprivate/OEBPS/Images/38_02.jpg?sign=1739222995-j4rSboe6NQkq5i6UV5qykrUrbx5fUyX3-0-cb1a79b6231b9c6eaa1484f35ea8a1aa)
2)定义一些超参数。
![](https://epubservercos.yuewen.com/B4645E/30403237107613906/epubprivate/OEBPS/Images/38_03.jpg?sign=1739222995-TV6ZIgssXQWW3qSRwqObUqACK0UpfgdE-0-ce8daeeb2c5afa1d88a3792437243bc8)
![](https://epubservercos.yuewen.com/B4645E/30403237107613906/epubprivate/OEBPS/Images/39_01.jpg?sign=1739222995-z1olJzb7wQRoL5wOcPHqk453LUyiDQ3U-0-59d960f7f2961cb293867b1782687570)
3)下载数据并对数据进行预处理。
![](https://epubservercos.yuewen.com/B4645E/30403237107613906/epubprivate/OEBPS/Images/39_02.jpg?sign=1739222995-hK455wrTkWxkFzZc3OUpN5A9cwx4rsYQ-0-6a51425c9339f0feeaf635b7d6fccd17)
说明:
● transforms.Compose可以把一些转换函数组合在一起。
● Normalize([0.5], [0.5])对张量进行归一化,这里两个0.5分别表示对张量进行归一化的全局平均值和方差。因图像是灰色的,则只有一个通道,如果有多个通道,需要有多个数字,如三个通道,应该是Normalize([m1,m2,m3], [n1,n2,n3])。
● download参数控制是否需要下载,如果./data目录下已有MNIST,可选择False。
● 用DataLoader得到生成器,可节省内存。
● torchvision及data为PyTorch的数据预处理工具。