newest

pages/Player_VS_AI.py

@@ -42,16 +42,16 @@ from const import (
     _MODEL_PATH
 )
 
-'''
-from ai import (
-    BOS_TOKEN_ID,
-    generate_gpt2,
-    load_model,
-)
-
-gpt2 = load_model()
-
-'''
+# '''
+# from ai import (
+#     BOS_TOKEN_ID,
+#     generate_gpt2,
+#     load_model,
+# )
+#
+# gpt2 = load_model()
+#
+# '''
 
 
 # Utils
@@ -415,6 +415,7 @@ def gomoku():
                 top_five_acts = [act for act, prob in sorted_acts_probs[:5]]
                 top_five_probs = [prob for act, prob in sorted_acts_probs[:5]]
         if response and session_state.ROOM.TURN == _BLACK:  # human turn
+            start_time = time.time()
             print("Your turn")
             # construction of clickable buttons
             cur_move = (session_state.ROOM.current_move // _BOARD_SIZE, session_state.ROOM.current_move % _BOARD_SIZE)
@@ -458,7 +459,8 @@ def gomoku():
                                 on_click=handle_click,
                                 args=(i, j),
                             )
-
+            end_time = time.time()
+            print("Time used for human move: ", end_time - start_time)
 
         elif response and session_state.ROOM.TURN == _WHITE:  # AI turn
             message.empty()