logo
Loading...

transfer learning_ResNet50_訓練時GPU記憶體空間不足怎麼辦? - Cupoy

老师您好~如题,我有在网络在找到部分解决办法,添加以下代码:# 我们为了防止碎片化和为了更好的利用内...

transfer learning,ResNet50

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了,之后就不能运行了,还有什么办法进一步解决我遇到的问题吗?谢谢!