Fixed seen for multu-GPU training

This commit is contained in:
AlexeyAB
2020-04-11 14:57:42 +03:00
parent 3d9aa2af47
commit 08b1c20e26

View File

@ -625,6 +625,7 @@ float train_networks(network *nets, int n, data d, int interval)
}
//cudaDeviceSynchronize();
*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)
{
printf("Syncing... ");