Function is_time_to_exit

BingBertSquad/utils.py:141–143

(args, epoch_steps=0, global_steps=0)

          .format(tag, micro_step, global_step, norm_groups))

def is_time_to_exit(args, epoch_steps=0, global_steps=0):
    return (epoch_steps >= args.max_steps_per_epoch) or \
            (global_steps >= args.max_steps)

def check_early_exit_warning(args):
    # Issue warning if early exit from epoch is configured