-
Notifications
You must be signed in to change notification settings - Fork 34
Open
Description
generate_data.py --street 1 --approximate root_nodes
in PyStack\src\NeuralNetwork\next_round_value.py
function _init_root_approximation_vars()
self.next_round_inputs = np.zeros([batch_size,BC,HC*PC + 1 + self.num_board_features], dtype=dt)
self.next_round_values = np.zeros([batch_size,BC,PC,HC], dtype=dt)
what is the batch_size?So next_round_inputs requires 22GB of memory。Do I need more than 60g of memory?
Metadata
Metadata
Assignees
Labels
No labels