def separate_bn_paras(modules):
时间: 2024-06-06 07:07:50 浏览: 138
rtw_efuse.rar_V2
5星 · 资源好评率100%
bn_paras = []
other_paras = []
for i in range(len(modules)):
module = modules[i]
if isinstance(module, torch.nn.BatchNorm2d):
bn_paras += list(module.parameters())
else:
other_paras += list(module.parameters())
return bn_paras, other_paras
阅读全文