摘要: 0.模块 torchvision有4个功能模块:model、datasets、transforms和utils。利用datasets可以下载一些经典数据集,本次笔记主要记录如何使用datasets的ImageFolder处理自定义数据集,以及如何使用transforms对源数据进行预处理、增强等。 阅读全文
posted @ 2020-12-09 16:48 wzdszh 阅读(223) 评论(0) 推荐(0)
摘要: 1.utils.data包括Dataset和DataLoader torch.utils.data.Dataset为抽象类,表示Dataset的抽象类,所有其他数据集都应该进行子类化,所有子类应该override,__len__和__getitem__,前者提供了数据集的大小,后者支持整数索引,范围 阅读全文
posted @ 2020-12-09 16:34 wzdszh 阅读(205) 评论(0) 推荐(0)
摘要: 代码中的变量定义皆是同前面笔记相同 1.PyTorch的nn模块 神经网络的结构与原理相信大家已经非常熟悉,这里不再赘述。 PyTorch有一个专门用于神经网络的完整子模块:torch.nn。该子模块包含创建各种神经网络体系结构所需的构建块。这些构建块在PyTorch术语中称为module(模块), 阅读全文
posted @ 2020-12-09 15:56 wzdszh 阅读(180) 评论(0) 推荐(0)
摘要: 1.一个线性回归的例子 假设你去了一些鲜为人知的地方旅游,然后带回了一个花哨的壁挂式模拟温度计。这个温度计看起来很棒,非常适合你的客厅。唯一的缺点是它不显示单位。不用担心,你有一个计划。你用自己喜欢的单位建立一个读数和相应温度值的数据集,然后选择一个模型,并迭代调整单位的权重,直到误差的测量值足够低 阅读全文
posted @ 2020-12-09 15:25 wzdszh 阅读(242) 评论(0) 推荐(0)