another compile fix

SpongeBab · Mar 11, 2020 · 56bd298 · 56bd298
1 parent 374c769
commit 56bd298
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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);