cpickle.picklingerror can't pickle type 'module' attribute lookup __builtin__.module failed

最近在使用进程池的时候发现了这错误!下面是自己调用的简化过程,不考虑语法错误,大概就是map函数的第二个参数中放了一个包含类实例的迭代器,这个类包含了一个模型参数初始化的tensorflow的session

def func(net):
return net.predict()
 
class Detecter(object):
def __init__(self, file):
with tf.Graph().as_default():
self.sess = tf.Session()
tf.train.Saver(self.sess,file)
def predict(self):
self.sess.run()
 
detecter = [Detecter('./model.ckpt')]*4
for batch in range(0,4,4):
bb = multiprocessing.pool.map(func, izip(detecter[:4]))

解决方案:将原先导入“from multiprocessing  import Pool”改为“from multiprocessing.pool import ThreadPool as Pool ”

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值