我是靠谱客的博主 轻松鲜花,最近开发中收集的这篇文章主要介绍pytorch, 遍历dataloade时,如何获取原始index,觉得挺不错的,现在分享给大家,希望可以做个参考。

概述

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所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(78)

评论列表共有 0 条评论

立即
投稿
返回
顶部