update pg、predict and add rl

2023-05-30 18:48:27 +08:00
parent 6abf9246c6
commit 41c70802fd
9 changed files with 488 additions and 2 deletions
@@ -86,6 +86,7 @@ class PolicyAgent(Agent):

    @classmethod
    def load(cls, path, encoder):
-        model = AlphaGoModel(encoder.get_input_shape(), is_policy_net=True).to('cuda')
+        device = 'cuda' if torch.cuda.is_available() else 'cpu'
+        model = AlphaGoModel(encoder.get_input_shape(), is_policy_net=True).to(device)
        model.load_state_dict(torch.load(path))
        return cls(model, encoder)