概述
pytorch,遍历dataloader的时候
train_loader = DataLoader(train_set, batch_size=batch_size_train, shuffle=True)
因为dataloader已经被shuffle过,
如何知道,他在原始dataset中的
index
torch.utils.data — PyTorch 1.9.1 documentation
最后
以上就是轻松鲜花为你收集整理的pytorch, 遍历dataloade时,如何获取原始index的全部内容,希望文章能够帮你解决pytorch, 遍历dataloade时,如何获取原始index所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复