input encoding

demo/interfaces/activations.py

@@ -113,7 +113,8 @@ with gr.Blocks() as interface:
                     choices=[
                         ("classical", InputEncoding.INPUT_CLASSICAL_112_PLANE), 
                         ("repeated", InputEncoding.INPUT_CLASSICAL_112_PLANE_REPEATED), 
-                        ("no history", InputEncoding.INPUT_CLASSICAL_112_PLANE_NO_HISTORY)
+                        ("no history repeated", InputEncoding.INPUT_CLASSICAL_112_PLANE_NO_HISTORY_REPEATED),
+                        ("no history zeros", InputEncoding.INPUT_CLASSICAL_112_PLANE_NO_HISTORY_ZEROS)
                     ],
                     value=InputEncoding.INPUT_CLASSICAL_112_PLANE,
                 )
@@ -184,6 +185,6 @@ with gr.Blocks() as interface:
     )
     channel_index.change(
         render_activations,
-        inputs=[board, activations, layer_index, channel_index],
+        inputs=[board, activations, input_encoding, layer_index, channel_index],
         outputs=[image_board, colorbar],
     )

demo/interfaces/encodings.py

@@ -33,7 +33,7 @@ def make_render(game_pgn:str, board_fen:str, input_encoding:InputEncoding, plane
     return board, *make_board_plot(board, input_encoding, plane_index)
 
 def make_board_plot(board:LczeroBoard, input_encoding:InputEncoding, plane_index:int):
-    input_tensor = board.to_input_tensor(input_encoding)
+    input_tensor = board.to_input_tensor(input_encoding=input_encoding)
     board.render_heatmap(
         input_tensor[plane_index].view(64),
         save_to=f"{FIGURE_DIRECTORY}/encodings.svg",
@@ -65,7 +65,8 @@ with gr.Blocks() as interface:
                     choices=[
                         ("classical", InputEncoding.INPUT_CLASSICAL_112_PLANE), 
                         ("repeated", InputEncoding.INPUT_CLASSICAL_112_PLANE_REPEATED), 
-                        ("no history", InputEncoding.INPUT_CLASSICAL_112_PLANE_NO_HISTORY)
+                        ("no history repeated", InputEncoding.INPUT_CLASSICAL_112_PLANE_NO_HISTORY_REPEATED),
+                        ("no history zeros", InputEncoding.INPUT_CLASSICAL_112_PLANE_NO_HISTORY_ZEROS)
                     ],
                     value=InputEncoding.INPUT_CLASSICAL_112_PLANE,
                 )

demo/interfaces/gradients.py

@@ -116,7 +116,8 @@ with gr.Blocks() as interface:
                     choices=[
                         ("classical", InputEncoding.INPUT_CLASSICAL_112_PLANE), 
                         ("repeated", InputEncoding.INPUT_CLASSICAL_112_PLANE_REPEATED), 
-                        ("no history", InputEncoding.INPUT_CLASSICAL_112_PLANE_NO_HISTORY)
+                        ("no history repeated", InputEncoding.INPUT_CLASSICAL_112_PLANE_NO_HISTORY_REPEATED),
+                        ("no history zeros", InputEncoding.INPUT_CLASSICAL_112_PLANE_NO_HISTORY_ZEROS)
                     ],
                     value=InputEncoding.INPUT_CLASSICAL_112_PLANE,
                 )
@@ -201,6 +202,6 @@ with gr.Blocks() as interface:
     for render_arg in [average_over_planes, begin_average_index, end_average_index, plane_index]:
         render_arg.change(
             render_gradients,
-            inputs=[board, gradients, average_over_planes, begin_average_index, end_average_index, plane_index],
+            inputs=[board, gradients, input_encoding, average_over_planes, begin_average_index, end_average_index, plane_index],
             outputs=[image_board, colorbar],
         )

requirements.txt

@@ -1,2 +1,2 @@
-lczerolens[viz]>=0.3.3
+lczerolens[viz] @ git+https://github.com/Xmaster6y/lczerolens.git@04fc00d
 gdown