some try for policy_gradient

2023-06-01 19:00:56 +08:00
parent be222b4c82
commit 5ccfbeb4e8
5 changed files with 295 additions and 40 deletions
@@ -133,3 +133,7 @@ class AlphaGoEncoder(Encoder):

    def shape(self):
        return self.num_planes, self.board_height, self.board_width
+
+
+def create(board_size):
+    return AlphaGoEncoder(board_size)