Alexey
2018-01-14 270a38ecad42423053a9ca9a249d0763144fc1b9
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;
    }
}