tensorflow在训练过程中出现内存溢出的问题
最近在使用perceptual_loss过程中，出现了模型随着训练占用的内存越来越大的问题，千万不要在遍历数据的过程中使用tf的相关操作，必要的操作放到模型中，如果在模型外执行，会不断增加新的节点，导致网络占用的内存随着训练的进行越来越大...
·
最近在使用perceptual_loss过程中，出现了模型随着训练占用的内存越来越大的问题，千万不要在遍历数据的过程中使用tf的相关操作，必要的操作放到模型中，如果在模型外执行，会不断增加新的节点，导致网络占用的内存随着训练的进行越来越大
汇聚全球AI编程工具，助力开发者即刻编程。
更多推荐
0
0- 0
已为社区贡献1条内容
所有评论(0)