training-data-analyst/self-paced-labs/vertex-ai/vertex-ai-qwikstart\lab_exercise.ipynb

[dsgogogo]

Ln 27 - STEPS_PER_EPOCH = (TOTAL_TRAIN_EXAMPLES // (BATCH_SIZE*N_CHECKPOINTS))

I think the steps should be as below. We don't need checkpoints in the formula.

STEPS_PER_EPOCH = (TOTAL_TRAIN_EXAMPLES // (BATCH_SIZE)