added way to skip batches based on Peters previous PR

Created by: ngoyal2707

Based on https://github.com/facebookresearch/metaseq/pull/500

that version was triggering model save every update.