diff options
Diffstat (limited to 'include/nn.h')
| -rw-r--r-- | include/nn.h | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/include/nn.h b/include/nn.h index 871bd0f..3a2277a 100644 --- a/include/nn.h +++ b/include/nn.h @@ -31,7 +31,7 @@ // somewhat internal, important for generating training data void nn_cell_idxs(uint8_t *out, tak_state_p state); -void nn_prepare_head(float out_head[3], tak_state_p state); +void nn_prepare_head(float out_head[2], tak_state_p state); void nn_prepare_input(float *cur_board, tak_state_p state, const float embedding[NN_N_EMB]); @@ -41,10 +41,10 @@ int nn_last_cell_idxs(uint8_t *out, tak_state_p state); // Network // =================================================================== -float eval_net(const float *board, const unsigned int n_inp, - const float *w, const float *biases, - const float *logit_w, const float logit_b, - const unsigned int n_dense); +float eval_net(const float *board, const unsigned int n_inp, const float *w, + const float *biases, const float *logit_w, const float logit_b, + const unsigned int n_dense, + const float parity); // Evaluators for the two built-in networks. float five_nn_fn(tak_state_p state); |