Created by: baiyfbupt
我有部分参数params和其对应的梯度grads,优化器optimizer,现在需要根据梯度grads对参数进行优化,在静态图中这样实现:
optimizer.apply_gradients([params, grads])
在动态图中用apply_gradients
或者apply_optimize
接口会报错,想知道是否有其他方式满足需求?
例如pytorch中可以用以下方式实现
param.grad = gradx
optimizer.step()