def get_inner_model()

in src/train.py [0:0]

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

def get_inner_model(model):
    #return model.module if isinstance(model, DataParallel) else model
    return model.module if isinstance(model, DistributedDataParallel) else model