def METHOD_NAME(args, colbert, optimizer, batch_idx, savepath=None, consumed_all_triples=False):
    # arguments = dict(args)
    # arguments = args.input_arguments.__dict__
    # TODO: Call provenance() on the values that support it??
    checkpoints_path = savepath or os.path.join(Run().path_, 'checkpoints')
    name = None
    try:
        save = colbert.save
    except:
        save = colbert.module.save
    if not os.path.exists(checkpoints_path):
        os.makedirs(checkpoints_path)
    path_save = None
    if consumed_all_triples or (batch_idx % 2000 == 0):
        # name = os.path.join(path, "colbert.dnn")
        # save_checkpoint(name, 0, batch_idx, colbert, optimizer, arguments)
        path_save = os.path.join(checkpoints_path, "colbert")
    if batch_idx in SAVED_CHECKPOINTS:
        # name = os.path.join(path, "colbert-{}.dnn".format(batch_idx))
        # save_checkpoint(name, 0, batch_idx, colbert, optimizer, arguments)
        path_save = os.path.join(checkpoints_path, f"colbert-{batch_idx}")
    if path_save:
        print(f"#> Saving a checkpoint to {path_save} ..")
        # checkpoint = {}
        # checkpoint['batch'] = batch_idx
        # checkpoint['epoch'] = 0
        # checkpoint['model_state_dict'] = model.state_dict()
        # checkpoint['optimizer_state_dict'] = optimizer.state_dict()
        # checkpoint['arguments'] = arguments
        save(path_save)
    return path_save