Compile fix

pull/4976/head
AlexeyAB 5 years ago
parent 5e54523960
commit 374c769062
  1. 2
      src/connected_layer.h

@ -22,7 +22,7 @@ void statistics_connected_layer(layer l);
#ifdef GPU #ifdef GPU
void forward_connected_layer_gpu(connected_layer layer, network_state state); void forward_connected_layer_gpu(connected_layer layer, network_state state);
void backward_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 push_connected_layer(connected_layer layer);
void pull_connected_layer(connected_layer layer); void pull_connected_layer(connected_layer layer);
#endif #endif

Loading…
Cancel
Save