你好，使用VGG模型运行grafting_cifar时，出现错误

[linsencc]

models的vgg代码中构造网络时使用了 _make_layers构造卷积层，在运行grafting.py时会出现UnboundLocalError: local variable 'w' referenced before assignment

[fxmeng]

你在函数开始的时候给w一个初始值就好了

[linsencc]

可能不行，代码中根据‘conv’ 字符串查找卷积层，但是这里vgg用了_make_layers将参数层加入到features序列中 ，导致key中没有‘conv’， 只有‘features’ ··· def grafting(net, epoch): …… for i, (key, u) in enumerate(net.state_dict().items()): if 'conv' in key: w = round(args.a / np.pi * np.arctan(args.c * (entropy(u) - entropy(checkpoint[key]))) + 0.5, 2) model[key] = u * w + checkpoint[key] * (1 - w) net.load_state_dict(model) ···

[fxmeng]

哦哦，那这里需要改一下判断语句了，比如：if len(n.shape)==4