hanxuanyan001的博客

原创 pytorch :: Dataloader中的迭代器和生成器应用

在使用pytorch训练模型，经常需要加载大量图片数据，因此pytorch提供了好用的数据加载工具Dataloader。 为了实现小批量循环读取大型数据集，在Dataloader类具体实现中，使用了迭代器和生成器。 这一应用场景正是python中迭代器模式的意义所在，因此本文对Dataloader中代码进行解读，可以更好的理解python中迭代器和生成器的概念。 本文的内容主要有： 解释pytho...