src/softmax_layer_kernels.cu
@@ -42,7 +42,7 @@ extern "C" void backward_softmax_layer_gpu(const softmax_layer layer, network_state state) { copy_ongpu(layer.batch*layer.inputs, layer.delta_gpu, 1, state.delta, 1); axpy_ongpu(layer.batch*layer.inputs, 1, layer.delta_gpu, 1, state.delta, 1); } /* This is if you want softmax w/o log-loss classification. You probably don't.