Method load_weights

pytorch/rl_trader.py:109–110

(self, path)

    torch.save(self.state_dict(), path)

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