diff options
Diffstat (limited to 'src/testNeuralNetwork.cpp')
-rw-r--r-- | src/testNeuralNetwork.cpp | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/testNeuralNetwork.cpp b/src/testNeuralNetwork.cpp index 3e1640e..c60f9c5 100644 --- a/src/testNeuralNetwork.cpp +++ b/src/testNeuralNetwork.cpp @@ -101,7 +101,11 @@ int main (int argc, char *argv[]) prop.fProp(minibatch.topRows(myParam.ngram_size-1)); // Do full forward prop through output word embedding layer - prop.output_layer_node.param->fProp(prop.second_hidden_activation_node.fProp_matrix, scores); + if (prop.skip_hidden) + prop.output_layer_node.param->fProp(prop.first_hidden_activation_node.fProp_matrix, scores); + else + prop.output_layer_node.param->fProp(prop.second_hidden_activation_node.fProp_matrix, scores); + // And softmax and loss double minibatch_log_likelihood; |