i try to train gomoku. but i got an error about muzero-general HOT 4 CLOSED

### Training
self.results_path = os.path.join(os.path.dirname(__file__), "../results", os.path.basename(__file__)[:-3], datetime.datetime.now().strftime("%Y-%m-%d--%H-%M-%S"))  # Path to store the model weights and TensorBoard logs
self.training_steps = 50000  # Total number of training steps (ie weights update according to a batch)
self.batch_size = 512  # Number of parts of games to train on at each training step
self.checkpoint_interval = 10  # Number of training steps before using the model for sef-playing
self.value_loss_weight = 0.25  # Scale the value loss to avoid overfitting of the value function, paper recommends 0.25 (See paper appendix Reanalyze)
self.training_device = "cuda" if torch.cuda.is_available() else "cpu"  # Train on GPU if available

self.optimizer = "SGD"  # "Adam" or "SGD". Paper uses SGD
self.weight_decay = 1e-4  # L2 weights regularization
self.momentum = 0.9  # Used only if optimizer is SGD

### Self-Play
self.num_actors = 4  # Number of simultaneous threads self-playing to feed the replay buffer
self.max_moves = 121  # Maximum number of moves if game is not finished before
self.num_simulations = 121  # Number of future moves self-simulated
self.discount = 1  # Chronological discount of the reward
self.temperature_threshold = 80  # Number of moves before dropping temperature to 0 (ie playing according to the max)