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