Merge pull request #5145 from keita-honsho/fix_errors

fix some errors.
pull/5181/head
Alexey 5 years ago committed by GitHub
commit cedfd792c6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 2
      src/network_kernels.cu

@ -246,7 +246,7 @@ void backward_network_gpu(network net, network_state state)
if (net.adversarial) {
int x_size = get_network_input_size(net)*net.batch;
printf(" x_size = %d, original_delta = %p, original_input = %p, net.learning_rate = %d \n",
x_size, original_delta, original_input, x_size, net.learning_rate);
x_size, original_delta, original_input, net.learning_rate);
axpy_ongpu(x_size, net.learning_rate, original_delta, 1, original_input, 1);
constrain_min_max_ongpu(x_size, 0, 1, original_input, 1);
}

Loading…
Cancel
Save