最近在使用perceptual_loss过程中,出现了模型随着训练占用的内存越来越大的问题,千万不要在遍历数据的过程中使用tf的相关操作,必要的操作放到模型中,如果在模型外执行,会不断增加新的节点,导致网络占用的内存随着训练的进行越来越大
tensorflow在训练过程中出现内存溢出的问题
最新推荐文章于 2022-04-16 13:28:45 发布
最近在使用perceptual_loss过程中,出现了模型随着训练占用的内存越来越大的问题,千万不要在遍历数据的过程中使用tf的相关操作,必要的操作放到模型中,如果在模型外执行,会不断增加新的节点,导致网络占用的内存随着训练的进行越来越大