def train()

in opacus_lab/models/GPT2/train.py [0:0]

• 16 lines of code
• 1 McCabe index (conditional complexity)

def train(
    model,
    device,
    train_loader,
    val_loader,
    epoch,
    optimizer,
    virtual_batch_size,
    max_iters,
    scheduler,
    val_freq=2048,
    val_iters=512,
    print_freq=100,
    delta=1.0,
    Huggingface=False,
    checkpoint_path=None,