tensorflow 多模型模型切换释放显存

#模型定义
class Model1(object):
    def __init__(self,cfg):
        self.sess = InteractiveSession(...)




#主程序运行部分
M1 = Model1(cfg1)
M1.pred()

#现在想使用另一个模型 防止out of memory  可先释放第一个模型分配的显存再初始化第二个模型

yolo1.session.close()
tf.reset_default_graph()

#就是使用close()释放掉模型中的session部分,session里面实现分配现存的。
#直接使用del M1 是不能释放掉显存的(主程序运行时)

#初始化另一个模型
M2 = Model2(cfg2)



主要就是对模型里面的session使用close()方法

上一篇:阿里云服务器搭建wordpress博客


下一篇:开放下载!《阿里语音与信号处理技术》精选集