#模型定义
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()方法