def convert_type()

in utils/fid.py [0:0]

• 8 lines of code
• 7 McCabe index (conditional complexity)

def convert_type(features, cuda):
    if cuda and (type(features) == np.ndarray):
        features = torch.from_numpy(features).cuda()
    elif (not cuda) and (type(features) != np.ndarray):
        features = features.cpu().numpy()
    elif cuda and (features.device.type == 'cpu'):
        features = features.cuda()
    return features