为什么定义的resnet网络训练时gpu显存占用这么大啊？

Eric-mingjie / network-slimming

Network Slimming (Pytorch) (ICCV 2017)

MIT License

907 stars 214 forks source link

Closed LW-CVer closed 3 years ago

LW-CVer commented 3 years ago

batch=32 depth=20 其实就20层左右的卷积，结果显存占用8G，官方resnet50都只占6.5G显存网络定义跟官方定义方式也差不多啊，为什么显存占用会这么大啊 @ @Eric-mingjie