||
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!
Archiver|手机版|科学网 ( 京ICP备07017567号-12 )
GMT+8, 2024-11-24 08:57
Powered by ScienceNet.cn
Copyright © 2007- 中国科学报社