手动更新参数会建立计算图吗?

时间:2019-07-16 01:31:36

标签: pytorch

以下更新方法是否可以在Pytorch中构建计算图?

   ...

   grad = torch.autograd.grad(loss, model.parameters())
   for l, param in enumerate(model.parameters()):
       param -= learning_rate * grad[l] # does this line of code build computational graph?

   ...

如果这不能建立计算图,那么我是否想知道如何实现像MAML(模型不可知的元学习)中的双向向后传递的实现?

0 个答案:

没有答案