def METHOD_NAME(e):
    wps = e.state.metrics["wps"]
    # Skip iterations 2, 3, 4 if backend is Horovod on CUDA,
    # wps is abnormally low for these iterations
    # otherwise, other values of wps are OK
    if idist.model_name() == "horovod-dist" and e.state.iteration in (2, 3, 4):
        return
    low_wps = estimated_wps * lower_bound_factor
    high_wps = estimated_wps * upper_bound_factor
    assert low_wps < wps <= high_wps, f"{e.state.iteration}: {low_wps} < {wps} <= {high_wps}"