From 624da7dd62c931215bd845e44d755160a3e6d3dd Mon Sep 17 00:00:00 2001 From: Martin Eyben Date: Mon, 12 May 2025 13:35:27 +0200 Subject: [PATCH] fix: torch parsing --- go_ia/main.py | 1 + 1 file changed, 1 insertion(+) diff --git a/go_ia/main.py b/go_ia/main.py index 146bee2..83130f2 100755 --- 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="" )