feat: add a profiled predict mode

Author: Sigmanificient

src/analyzer/__init__.py

@@ -1,12 +1,13 @@
 from .data_loader import load_chessfile_predict, load_chessfile_train
 from .labels import label_to_vector, vector_to_label
-from .modes import predict_mode, train_mode
+from .modes import predict_mode, predict_mode_profiled, train_mode
 
 __all__ = [
     "load_chessfile_predict",
     "load_chessfile_train",
     "label_to_vector",
     "vector_to_label",
     "predict_mode",
+    "predict_mode_profiled",
     "train_mode",
 ]

src/analyzer/modes.py

@@ -6,6 +6,7 @@
 
 from .data_loader import load_chessfile_predict, load_chessfile_train
 from .labels import vector_to_label
+from .profile import profile_it
 
 
 def process_fen(network: Network, encoding: str, fen: str):
@@ -30,6 +31,18 @@ def predict_mode(
             print(result)
 
 
+@profile_it
+def predict_mode_profiled(
+    network: Network, chessfile: str, encoding: str = "simple"
+) -> None:
+    fens = load_chessfile_predict(chessfile)
+
+    dispatch = functools.partial(process_fen, network, encoding)
+
+    for c, fen in enumerate(fens):
+        print(c, dispatch(fen))
+
+
 def train_mode(
     network: Network,
     chessfile: str,

src/analyzer/profile.py

@@ -0,0 +1,13 @@
+import cProfile
+from pstats import SortKey, Stats
+
+
+def profile_it(func):
+    def wrapper(*args, **kwargs):
+        with cProfile.Profile() as pr:
+            result = func(*args, **kwargs)
+
+        Stats(pr).sort_stats(SortKey.CUMULATIVE).print_stats()
+        return result
+
+    return wrapper

src/my_torch/tools/my_torch_analyzer.py

@@ -1,8 +1,9 @@
 #!/usr/bin/env python3
 import argparse
 import sys
+import time
 
-from analyzer import predict_mode, train_mode
+from analyzer import predict_mode, predict_mode_profiled, train_mode
 from my_torch import Network
 
 
@@ -52,6 +53,10 @@ def main():
         help="Board encoding method (default: simple)",
     )
 
+    parser.add_argument(
+        "--profiled", action="store_true", help=argparse.SUPPRESS
+    )
+
     args = parser.parse_args()
 
     try:
@@ -66,7 +71,14 @@ def main():
         return 84
 
     if args.predict:
-        predict_mode(network, args.chessfile, encoding=args.encoding)
+        if args.profiled:
+            print("Using profiled version, single threaded.")
+            time.sleep(1)
+            predict_mode_profiled(
+                network, args.chessfile, encoding=args.encoding
+            )
+        else:
+            predict_mode(network, args.chessfile, encoding=args.encoding)
     else:
         savefile = args.save if args.save else args.loadfile
         train_mode(