
pytorch
可达鸭Ya。
这个作者很懒,什么都没留下…
展开
-
使用PyTorch框架实现CIFAR-10图像分类模型的训练和评估
最重要的部分是模型设计部分,建议先在纸上计算一个批次数据通过卷积、池化等部分变化的整个流程,然后再动手写代码会快很多。原创 2024-10-20 00:00:52 · 286 阅读 · 0 评论 -
DataLoader使用的Map-style datasets 和 Iterable-style datasets的区别
Map-style datasets是一个可以通过索引来访问数据的集合,定义的类中要实现__len__()和__getitem__()这两种方法。Iterable-style datasets不能通过索引访问,只能通过迭代方式逐个访问,定义的类中要实现__iter()__方法。Iterable-style datasets是动态生成、大小不固定、不支持随机访问不支持shuffle操作的数据集。Map-style datasets是静态存储、大小固定、随机访问且支持shuffle操作的数据集。原创 2024-09-11 21:12:24 · 327 阅读 · 0 评论