深度学习
关注
分享
扫一扫
文章平均质量分 97
pytorch等学习
早点打完工去睡觉
基础小白
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
Pytorch（二）
DataLoadr是Pytorch中数据读取的一个重要的接口，该接口的目的：将自定义的Dataset根据batch_size大小，是否shuffle等封装成一个Batch_Size大小的Tensor，用于后面的训练。DataLoader(object)的参数dataset(Dataset): 传入的数据集batch_size(int, optional): 每个batch有多少个样本shuffle(bool, optional): 在每个epoch开始的时候，对数据进行重新排序。原创 2024-11-07 11:16:14 · 1343 阅读 · 0 评论 -
Pytorch（一）
Pytorch学习原创 2024-10-29 23:18:00 · 1038 阅读 · 0 评论