transfer learning_ResNet50_訓練時GPU記憶體空間不足怎麼辦?
2020/06/14 01:56 PM
電腦視覺深度學習論壇
骑士
觀看數：0
回答數：4
收藏數：0
transfer learning
ResNet50
老师您好~
如题，我有在网络在找到部分解决办法，添加以下代码：
# 我们为了防止碎片化和为了更好的利用内存，tensorflow初始化时会默认占用全部显存https://blog.csdn.net/tsyccnh/article/details/102938368
config = tf.compat.v1.ConfigProto() #allow_soft_placement=True
config.gpu_options.allow_growth = True #dynamically grow the memory used on the GPU
config.gpu_options.per_process_gpu_memory_fraction = 0.9 # 使用90%的显存
在Day021的作业练习中，我已经能够训练到Epoch39/100了，之后就不能运行了，还有什么办法进一步解决我遇到的问题吗？谢谢！