Skip to content

Exp-IBP

Exp-IBP entry point for the common expressive IBP trainer.

exp_ibp_train_model(*args, **kwargs)

Train Exp-IBP using the shared expressive IBP training procedure.

Arguments follow expressive_ibp_train_model; this entry point selects the method-specific loss while reusing its schedules and training loop.

Source code in CTRAIN/train/certified/exp_ibp.py 
 6
 7
 8
 9
10
11
12
def exp_ibp_train_model(*args, **kwargs):
    """Train Exp-IBP using the shared expressive IBP training procedure.

    Arguments follow expressive_ibp_train_model; this entry point selects the
    method-specific loss while reusing its schedules and training loop.
    """
    return expressive_ibp_train_model(*args, **kwargs, loss_func=get_exp_ibp_loss, method='exp_ibp')