Joseph Redmon
2015-03-24 56b6561ae4f1e238ba1a65701f91b40636037cc2
src/network_kernels.cu
@@ -20,8 +20,8 @@
#include "dropout_layer.h"
}
extern "C" float * get_network_output_gpu_layer(network net, int i);
extern "C" float * get_network_delta_gpu_layer(network net, int i);
float * get_network_output_gpu_layer(network net, int i);
float * get_network_delta_gpu_layer(network net, int i);
float *get_network_output_gpu(network net);
void forward_network_gpu(network net, network_state state)
@@ -196,8 +196,8 @@
    state.train = 1;
    forward_network_gpu(net, state);
    backward_network_gpu(net, state);
    if ((net.seen / net.batch) % net.subdivisions == 0) update_network_gpu(net);
    float error = get_network_cost(net);
    if ((net.seen / net.batch) % net.subdivisions == 0) update_network_gpu(net);
    return error;
}