使用tensorpack.dataflow.PrefetchData遇到的问题

2023年2月20日 506次阅读 来源: 一个三要不起

from tensorpack.dataflow.PrefetchData import PrefetchData

出现错误No module named ‘tensorpack.dataflow.PrefetchData’

通过官网API手册也没找到新版的PrefetchData使用例子，倒是看到了这么一句：
alias of MultiProcessPrefetchData

我想PrefetchData可能已经换名字了吧，然后拿着这个函数名MultiProcessPrefetchData在site-packages\tensorpack\dataflow 下.py文件中挨个挨个的找，终于让我找到了MultiProcessPrefetchData函数。

解决方案：

from tensorpack.dataflow.parallel import MultiProcessPrefetchData
用MultiProcessPrefetchData代替了PrefetchData

如果要应用到之前的代码中的话：
from tensorpack.dataflow.parallel import MultiProcessPrefetchData as PrefetchData

    原文作者：一个三要不起
    原文地址: https://www.jianshu.com/p/44e937d67208
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞