Fixed seen for multu-GPU training

pull/5817/head
AlexeyAB 5 years ago
parent 3d9aa2af47
commit 08b1c20e26
  1. 1
      src/network_kernels.cu

@ -625,6 +625,7 @@ float train_networks(network *nets, int n, data d, int interval)
} }
//cudaDeviceSynchronize(); //cudaDeviceSynchronize();
*nets[0].cur_iteration += (n - 1); *nets[0].cur_iteration += (n - 1);
*nets[0].seen = nets[0].batch * nets[0].subdivisions * get_current_iteration(nets[0]); // remove this line, when you will save to weights-file both: seen & cur_iteration
if (get_current_iteration(nets[0]) % interval == 0) if (get_current_iteration(nets[0]) % interval == 0)
{ {
printf("Syncing... "); printf("Syncing... ");

Loading…
Cancel
Save