def METHOD_NAME(exp_dir, epoch) -> None:
    model_path = get_checkpoint(exp_dir, epoch)
    for file_path in [
        model_path,
        get_optimizer_path(model_path),
        get_stats_path(model_path),
    ]:
        if os.path.isfile(file_path):
            logger.info("deleting %s" % file_path)
            os.remove(file_path)