自定义ModuleModule 是pytorch 组织神经网络的基本方式。 ... += int(mask.sum()) #根据mask的0\1进行获取保留了通道的 bn_module.weight.data.mul_(mask) mAP ...
確定! 回上一頁