attention-ocr icon indicating copy to clipboard operation
attention-ocr copied to clipboard

given num-epoch 200 but the training is continuing Step 328+, for how many steps it will continue?

Open anidiatm41 opened this issue 3 years ago • 1 comments

aocr train ./data/training.tfrecords --max-width 200 --max-height 100 --model-dir ./checkpoint --max-prediction 236 --num-epoch 200

anidiatm41 avatar Feb 04 '22 06:02 anidiatm41

Hi, I think I figured out the answer to this issue. There is this formula I found in stackoverflow:

(Number of images) / (Number of images per batch / step) = Number of steps per epoch

For example, in my case I used 3000 images, a batch-size of 50 and an epoch of 100. Using the formula I calculated 6,000 steps, which at the end was the true total amount of steps the model lasted training. Hope it helps.

akiradavid27 avatar Jun 21 '22 22:06 akiradavid27