From 374c769062b64c685cb673a0f5a2a3d3acef639d Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Wed, 11 Mar 2020 19:54:39 +0300 Subject: [PATCH] Compile fix --- src/connected_layer.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/connected_layer.h b/src/connected_layer.h index 80805cc2..284c5125 100644 --- 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