Method save_weights

pytorch/rl_trader.py:106–107

(self, path)

    return self.layers(X)

  def save_weights(self, path):
    torch.save(self.state_dict(), path)

  def load_weights(self, path):
    self.load_state_dict(torch.load(path))