AlphaZeroSimple
AlphaZeroSimple copied to clipboard
.cuda() -> .to(self.model.device)
https://github.com/JoshVarty/AlphaZeroSimple/blob/b68171a5cb9367b407017e07f5de3f65b10e888e/trainer.py#L83,L85