Joseph Redmon
2014-10-26 5c9a773bb6e994889354c181de5a872e867aa35b
src/convolutional_layer.c
@@ -212,7 +212,7 @@
{
    int size = layer.size*layer.size*layer.c*layer.n;
    axpy_cpu(layer.n, layer.learning_rate, layer.bias_updates, 1, layer.biases, 1);
    scal_cpu(layer.n,layer.momentum, layer.bias_updates, 1);
    scal_cpu(layer.n, layer.momentum, layer.bias_updates, 1);
    scal_cpu(size, 1.-layer.learning_rate*layer.decay, layer.filters, 1);
    axpy_cpu(size, layer.learning_rate, layer.filter_updates, 1, layer.filters, 1);
@@ -342,7 +342,7 @@
    check_error(cl);
}
//#define TIMEIT
#define TIMEIT
void forward_convolutional_layer_gpu(convolutional_layer layer, cl_mem in)
{
@@ -434,6 +434,12 @@
    cl_read_array(layer.biases_cl, layer.biases, layer.n);
}
void push_convolutional_layer(convolutional_layer layer)
{
    cl_write_array(layer.filters_cl, layer.filters, layer.c*layer.n*layer.size*layer.size);
    cl_write_array(layer.biases_cl, layer.biases, layer.n);
}
void update_convolutional_layer_gpu(convolutional_layer layer)
{
    int size = layer.size*layer.size*layer.c*layer.n;