freeze bn:
把所有相关的bn设置为 momentum=1.0 。...freeze 正常参数:
先比较两个state_dict,来freeze交集:
def freeze_model(model, defined_dict, keep_step=None):
for...:
pass_num += 1
return model, freezed_num, pass_num
之后再指定optimizer的时候要注意避开这部分参数,防止被freeze...p.requires_grad, model.parameters()), lr=0.001, \
betas=(0.9, 0.999), eps=1e-08, weight_decay=1e-5)
check freeze