for sample in iter(valdataloader):

aimerykong / predictive-filter-flow

Predictive Filter Flow for fully/self-supervised learning on various vision tasks

https://www.ics.uci.edu/~skong2/mgpff.html

140 stars 24 forks source link

Closed 394781865 closed 5 years ago

394781865 commented 5 years ago

查了pytorch文档，valdataloader是一个可迭代对象，用for循环遍历会报错。改成 dataiter = iter(dataloader) imgs, labels = next(dataiter) 就正常了。请问我的做法正确么？