tensorflow在训练过程中出现内存溢出的问题

最近在使用perceptual_loss过程中,出现了模型随着训练占用的内存越来越大的问题,千万不要在遍历数据的过程中使用tf的相关操作,必要的操作放到模型中,如果在模型外执行,会不断增加新的节点,导致网络占用的内存随着训练的进行越来越大


版权声明:本文为qq_37876289原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。