mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
Fixed seen for multu-GPU training
This commit is contained in:
@ -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... ");
|
||||
|
Reference in New Issue
Block a user