OrgLion-ML
/
darknet
mirror of https://github.com/AlexeyAB/darknet.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Compile fix

This commit is contained in:
AlexeyAB 2020-03-11 19:54:39 +03:00
parent 5e54523960
commit 374c769062
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/connected_layer.h
View File

@ -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