Compile fix

SpongeBab · Mar 11, 2020 · 374c769 · 374c769
1 parent 5e54523
commit 374c769
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/connected_layer.h b/src/connected_layer.h
@@ -22,7 +22,7 @@ void statistics_connected_layer(layer l);
 #ifdef GPU
 void forward_connected_layer_gpu(connected_layer layer, network_state state);
 void backward_connected_layer_gpu(connected_layer layer, network_state state);
-void update_connected_layer_gpu(connected_layer layer, int batch, float learning_rate, float momentum, float decay);
+void update_connected_layer_gpu(connected_layer layer, int batch, float learning_rate, float momentum, float decay, float loss_scale);
 void push_connected_layer(connected_layer layer);
 void pull_connected_layer(connected_layer layer);
 #endif