2020-7-13-将一个有特定功能的预训练网络加入另一个网络。

单独测试时,GPU内存占用很小,但加入另一个网络无法运行,GPU溢出。用排除法查出是 main.py 中cudnn.benchmark = True 导致。开关具体用法如下:benchmark的作用。下次GPU显存溢出可尝试修改。

你可能感兴趣的:(笔记,gpu)