我该如何解决向后（）得到意外的关键字参数'retain_variables'吗？

Question

我在下面编写了以下代码，但出现此错误：

TypeError: backward() got an unexpected keyword argument 'retain_variables'

我的代码是：

def learn(self, batch_state, batch_next_state, batch_reward, batch_action):
    outputs = self.model(batch_state).gather(1, batch_action.unsqueeze(1)).squeeze(1)
    next_outputs = self.model(batch_next_state).detach().max(1)[0]
    target = self.gamma*next_outputs + batch_reward
    td_loss = F.smooth_l1_loss(outputs, target)
    self.optimizer.zero_grad()
    td_loss.backward(retain_variables = True)
    self.optimizer.step()

Answer 1

为a_guest mentions in the comments:

  

应为keep_graph = True。

Answer 2

我遇到了同样的问题。这个解决方案对我有用。

td_loss.backward(retain_graph = True)

有效。