another compile fix

5 years ago · 56bd298297
parent 374c769062
commit 56bd298297
1 changed files with 1 additions and 1 deletions
--- a/src/local_layer.h
+++ b/src/local_layer.h
@ -15,7 +15,7 @@ extern "C" {
 #ifdef GPU
 void forward_local_layer_gpu(local_layer layer, network_state state);
 void backward_local_layer_gpu(local_layer layer, network_state state);
-void update_local_layer_gpu(local_layer layer, int batch, float learning_rate, float momentum, float decay);
+void update_local_layer_gpu(local_layer layer, int batch, float learning_rate, float momentum, float decay, float loss_scale);

 void push_local_layer(local_layer layer);
 void pull_local_layer(local_layer layer);