Tensorflow中因未释放计算图而导致显存爆炸运行越来越慢的问题
Tensorflow中因未释放计算图而导致显存爆炸运行越来越慢的问题
问题描述
在一次模型修改中,因为需要在每运行一个batch时,调用另外一个模型计算相关的embedding导入进来当前模型,所以导致了会出现类似以下的调用方式:
for batch in batches:
with tf.Session() as sess:
tfops = ...
原创
2019-08-16 23:45:27 ·
3600 阅读 ·
2 评论