【机器学习】pytorch中Dataloader的坑

基于pytorch训练一个模型,数据集size大概七十万 训练起来很慢 疯狂找缘由,后来通过一行行调试,发现是遍历Dataloader时太慢了 正在寻找解决办法。。(待更新2020.1.5)python 排坑1 :有大佬说PIL库比opencv慢,因而把代码里PIL库打开图片操做改为了opencvweb img = Image.open(img_name) #省略xxx行 t1 = time.ti
相关文章
相关标签/搜索