From 08b1c20e2666ce8813054cfed6243ad48feaac1d Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Sat, 11 Apr 2020 14:57:42 +0300 Subject: [PATCH] Fixed seen for multu-GPU training --- src/network_kernels.cu | 1 + 1 file changed, 1 insertion(+) diff --git a/src/network_kernels.cu b/src/network_kernels.cu index c611e229..68c2a208 100644 --- a/src/network_kernels.cu +++ b/src/network_kernels.cu @@ -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... ");