src/network_kernels.cu
@@ -78,6 +78,7 @@ void update_network_gpu(network net) { cuda_set_device(net.gpu_index); int i; int update_batch = net.batch*net.subdivisions; float rate = get_current_rate(net); @@ -377,7 +378,7 @@ float *get_network_output_layer_gpu(network net, int i) { layer l = net.layers[i]; cuda_pull_array(l.output_gpu, l.output, l.outputs*l.batch); if(l.type != REGION) cuda_pull_array(l.output_gpu, l.output, l.outputs*l.batch); return l.output; }
