From 56bd298297955b4ca80675374734158419c857a5 Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Wed, 11 Mar 2020 19:58:28 +0300 Subject: [PATCH] another compile fix --- src/local_layer.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/local_layer.h b/src/local_layer.h index 8ef0dbad..45e02a5c 100644 --- 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);