src/network_kernels.cu
@@ -51,7 +51,8 @@ fill_ongpu(l.outputs * l.batch, 0, l.delta_gpu, 1); } l.forward_gpu(l, state); cudaStreamSynchronize(get_cuda_stream()); if(net.wait_stream) cudaStreamSynchronize(get_cuda_stream()); state.input = l.output_gpu; } } @@ -113,6 +114,7 @@ state.truth = *net.truth_gpu; state.train = 1; forward_network_gpu(net, state); cudaStreamSynchronize(get_cuda_stream()); backward_network_gpu(net, state); }