<?xml version="1.0" encoding="utf-8" ?><rss version="2.0"><channel><title><![CDATA[TANHUIXIN_的博客]]></title><description><![CDATA[]]></description><link>https://blog.youkuaiyun.com/TANHUIXIN_</link><language>zh-cn</language><generator>https://blog.youkuaiyun.com/</generator><copyright><![CDATA[Copyright &copy; TANHUIXIN_]]></copyright><item><title><![CDATA[Pytorch GroupNorm引发的CUDA error: device-side assert triggered]]></title><link>https://blog.youkuaiyun.com/TANHUIXIN_/article/details/112672447</link><guid>https://blog.youkuaiyun.com/TANHUIXIN_/article/details/112672447</guid><author>TANHUIXIN_</author><pubDate>Fri, 15 Jan 2021 15:49:13 +0800</pubDate><description><![CDATA[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)的时候，报]]></description><category></category></item></channel></rss>