def weight_clipping(model, lower=-0.01, upper=0.01):
for params in model.params():
params_clipped = F.clip(params, lower, upper)
params.data = params_clipped.data
微信
公众号
扫码关注公众号