动态图resnet（模型结构有修改）训练时内存溢出，是否模型结构有问题，帮忙看一下代码？
Created by: qianledan
用官网给的动态图resnet模型，batch_size设置16，能够正常跑。
本人对该模型结构进行了部分的修改，训练时出现GPU内存溢出
Desktop.zip
Created by: qianledan
用官网给的动态图resnet模型，batch_size设置16，能够正常跑。
本人对该模型结构进行了部分的修改，训练时出现GPU内存溢出
Desktop.zip