AllSpark
AllSpark copied to clipboard
AllSpark中的InstanceNorm
感谢出彩的工作!
代码AllSpark中有一个InstanceNorm层
self.psi = nn.InstanceNorm2d(self.num_heads)
请问为什么用InstanceNorm以及为什么参数设置为num_heads呢?