def is_parallel()

in utils/utils.py [0:0]

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

def is_parallel(model):
    # Returns True if model is of type DP or DDP
    return type(model) in (nn.parallel.DataParallel, nn.parallel.DistributedDataParallel)