def refactor_transformer()

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

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

def refactor_transformer(
    GPT2, size="S", use_low_rank=False, lm_head_rank=768, dropout=0.0, perturb=False