2019年8月4日

pytorch之DataLoader()函数

摘要: 在训练神经网络时,最好是对一个batch的数据进行操作,同时还需要对数据进行shuffle和并行加速等。对此,PyTorch提供了DataLoader帮助我们实现这些功能。 DataLoader的函数定义如下: dataset:加载的数据集(Dataset对象) batch_size:batch s 阅读全文

posted @ 2019-08-04 15:28 赤子之心心有所属 阅读(5169) 评论(0) 推荐(1)

迭代器iter(db),db为实例化的类对象,类对象中没有__iter__,但有__getitem__

摘要: 在这种情况下用到了iter(),但类中也没有__iter__,所以有点疑惑 查阅资料后发现 当for发现没有__iter__但是有__getitem__的时候,会从0开始依次读取相应的下标,直到发生IndexError为止,这是一种旧的迭代协议。iter方法也会处理这种情况,在不存在__iter__ 阅读全文

posted @ 2019-08-04 11:45 赤子之心心有所属 阅读(433) 评论(0) 推荐(0)

导航