totensor和一些其他的

 

 compose组件

transforms.ToTensor()就是将pil图像或者numpy数据转为tensor,
并且高度,宽度,深度,在0-255之间的值,转为通道,高度,宽度,在0-1之间的值。
然后normalize就是对上面的tensor变量标准化。

 

输出=( 原始输入减去均值)/ 标准差

 

 

 这里的shuffle=true就是将batch里的数据随机打乱,一般都是true

 train=True就是导入的是训练集的图
num_workers=0在Windows下为0

 

 这里的iter()就是将测试集的数据转化为一个可以迭代的迭代器,

转化为迭代器后就可以用 .next()方法一次获取一批数据,包含图片和图片对应的标签

 

 

 

 这里.Adam(net.parameters(),lr=0.001),,net.parameters()的意思是训练net的所有参数



posted @   灵性  阅读(90)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· Ollama——大语言模型本地部署的极速利器
· 使用C#创建一个MCP客户端
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· ollama系列1:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
点击右上角即可分享
微信分享提示