Hi author, after I see your code, I think the number of iteration per epoch is decided by the number of image. Therefore, the program will cope with 5011 images per epoch in your code.
But in that paper, that author sets 70k iterations and batch size also equal to 2. So each epoch should deal with 3500 iterations and that means each epoch should process 7000 images rather that 5011.
I do not sure whether my thinking is right...
Hi author, after I see your code, I think the number of iteration per epoch is decided by the number of image. Therefore, the program will cope with 5011 images per epoch in your code. But in that paper, that author sets 70k iterations and batch size also equal to 2. So each epoch should deal with 3500 iterations and that means each epoch should process 7000 images rather that 5011. I do not sure whether my thinking is right...