fix: torch parsing

This commit is contained in:
Martin Eyben 2025-05-12 13:35:27 +02:00
parent 4e411c260d
commit 624da7dd62

View File

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