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

Distributed training

Browse Source
This commit is contained in:
Joseph Redmon
2014-12-07 00:41:26 -08:00
parent 1edcf73a73
commit 28e2115272
9 changed files with 229 additions and 110 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/connected_layer.c
View File

@ -112,12 +112,16 @@ void pull_connected_layer(connected_layer layer)
{
cl_read_array(layer.weights_cl, layer.weights, layer.inputs*layer.outputs);
cl_read_array(layer.biases_cl, layer.biases, layer.outputs);
cl_read_array(layer.weight_updates_cl, layer.weight_updates, layer.inputs*layer.outputs);
cl_read_array(layer.bias_updates_cl, layer.bias_updates, layer.outputs);
}
void push_connected_layer(connected_layer layer)
{
cl_write_array(layer.weights_cl, layer.weights, layer.inputs*layer.outputs);
cl_write_array(layer.biases_cl, layer.biases, layer.outputs);
cl_write_array(layer.weight_updates_cl, layer.weight_updates, layer.inputs*layer.outputs);
cl_write_array(layer.bias_updates_cl, layer.bias_updates, layer.outputs);
}
void update_connected_layer_gpu(connected_layer layer)