fix: torch parsing
This commit is contained in:
parent
4e411c260d
commit
624da7dd62
@ -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=""
|
||||||
)
|
)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user