diff options
Diffstat (limited to 'include/nn.c')
| -rw-r--r-- | include/nn.c | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/include/nn.c b/include/nn.c index 909b870..9b241a1 100644 --- a/include/nn.c +++ b/include/nn.c @@ -139,17 +139,17 @@ float eval_net(const float *board, const unsigned int n_inp, const float *w, float five_nn_fn(tak_state_p state) { static float cur_board[FIVE_INP_NUM] NN_ALIGNED; - nn_prepare_input(cur_board, state, five_embedding); - return eval_net(cur_board, FIVE_INP_NUM, &five_dense_weights[0][0], - five_dense_biases, five_logit_weights, five_logit_bias, + nn_prepare_input(cur_board, state, FIVE_embedding); + return eval_net(cur_board, FIVE_INP_NUM, &FIVE_dense_weights[0][0], + FIVE_dense_biases, FIVE_logit_weights, FIVE_logit_bias, FIVE_DENSE_NUM, state->ply & 1 ? 1.0f : -1.0f); } float six_nn_fn(tak_state_p state) { static float cur_board[SIX_INP_NUM] NN_ALIGNED; - nn_prepare_input(cur_board, state, six_embedding); - return eval_net(cur_board, SIX_INP_NUM, &six_dense_weights[0][0], - six_dense_biases, six_logit_weights, six_logit_bias, + nn_prepare_input(cur_board, state, SIX_embedding); + return eval_net(cur_board, SIX_INP_NUM, &SIX_dense_weights[0][0], + SIX_dense_biases, SIX_logit_weights, SIX_logit_bias, SIX_DENSE_NUM, state->ply & 1 ? 1.0f : -1.0f); } |