def set_decode_type()

in src/train.py [0:0]

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

def set_decode_type(model, decode_type):
    #if isinstance(model, DataParallel):
    if isinstance(model, DistributedDataParallel):
        model = model.module
    model.set_decode_type(decode_type)