NJU1healer的个人博客分享 http://blog.sciencenet.cn/u/NJU1healer

博文

Pytorch采样器详解--结合DataLoader使用

已有 1605 次阅读 2020-11-9 16:21 |个人分类:Pytorch|系统分类:科研笔记

Pytorch采样器详解--结合DataLoader使用,产生选取样本的随机编号,用于抽样!!

traindataloader = torch.utils.data.DataLoader(dataset=traindataset, sampler=RandomSampler(traindataset),

                                                  batch_size=args.batchsize, num_workers=args.workers)

testdataloader = torch.utils.data.DataLoader(dataset=testdataset, sampler=SequentialSampler(testdataset),

                                                 batch_size=args.batchsize, num_workers=args.workers)

具体参数详解见博客:

https://www.cnblogs.com/marsggbo/p/11541054.html

点滴分享,福泽你我!Add oil!



http://wap.sciencenet.cn/blog-3428464-1257685.html

上一篇:Scipy randint 与 Numpy randint 的区别
下一篇:池化总结

0

该博文允许注册用户评论 请点击登录 评论 (0 个评论)

数据加载中...
扫一扫,分享此博文

Archiver|手机版|科学网 ( 京ICP备07017567号-12 )

GMT+8, 2021-11-29 08:48

Powered by ScienceNet.cn

Copyright © 2007- 中国科学报社

返回顶部