fix: torch parsing

2025-05-12 13:35:27 +02:00 · 2025-05-12 13:35:27 +02:00 · 624da7dd62
commit 624da7dd62
parent 4e411c260d
1 changed files with 1 additions and 0 deletions
--- a/go_ia/main.py
+++ b/go_ia/main.py
@ -256,6 +256,7 @@ def setup_device():


 if __name__ == "__main__":
+    torch.serialization.add_safe_globals([GoDataset])
    parser = argparse.ArgumentParser(
        prog="alphaChadGo", description="a random ai evaluating go", epilog=""
    )