【发布时间】:2020-11-04 14:06:33
【问题描述】:
我已经使用 tensorflow 训练了一个 SRGAN 模型。但是在预测过程中,该模型需要大量内存(RAM),专门用于较大的图像。如何减少预测期间消耗的内存?
【问题讨论】:
标签: tensorflow memory-management
我已经使用 tensorflow 训练了一个 SRGAN 模型。但是在预测过程中,该模型需要大量内存(RAM),专门用于较大的图像。如何减少预测期间消耗的内存?
【问题讨论】:
标签: tensorflow memory-management
尝试将图像分成块,处理然后合并。
【讨论】: