mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

better multigpu

Browse Source
This commit is contained in:
Joseph Redmon
2016-09-20 11:34:49 -07:00
parent 5c067dc447
commit 73f7aacf35
26 changed files with 404 additions and 200 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/network.h
View File

@ -65,7 +65,8 @@ typedef struct network_state {
} network_state;
#ifdef GPU
float train_networks(network *nets, int n, data d);
float train_networks(network *nets, int n, data d, int interval);
void sync_nets(network *nets, int n, int interval);
float train_network_datum_gpu(network net, float *x, float *y);
float *network_predict_gpu(network net, float *input);
float * get_network_output_gpu_layer(network net, int i);