1 files changed, 1 insertions, 1 deletions
diff --git a/src/nn.h b/src/nn.h
index 4a47e63..88b4462 100644
--- a/src/nn.h
+++ b/src/nn.h
@@ -15,7 +15,7 @@ typedef struct Layer {
     size_t neurons, input_nodes;
 } Layer;
 
-void nn_layer_init_weights(Layer *layer, size_t nmemb, size_t input_size);
+void nn_layer_init_weights(Layer *layer, size_t nmemb, size_t input_cols);
 void nn_layer_free_weights(Layer *layer, size_t nmemb);
 
 double * nn_layer_forward(Layer layer, double *input, size_t input_shape[2]); //TODO