- 博客(1)
- 收藏
- 关注
原创 Pytorch GroupNorm引发的CUDA error: device-side assert triggered
Pytorch GroupNorm引发的CUDA error: device-side assert triggered 最近在尝试用Yolov3做检测,因为GPU消耗的问题只能设置batchsize=8,担心由于batchsize过小而造成BatchNormalization效果不好,故想改用GroupNormalization。 然而神奇的事情发生了,nn.GroupNorm(num_groups=4, num_channels=filters, eps=1e-05, affine=True)的时候,报
2021-01-15 15:49:13 232
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人