AlexeyAB
2018-01-07 aeb15b3cb9157f5d0b2a9962e17de22560b8a1b2
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;
    }