a02_zoo.py 文件源码

def get_random_state(cnn_type):
    if cnn_type == 'VGG16' or cnn_type == 'VGG16_DROPOUT':
        return 51
    elif cnn_type == 'VGG19_KERAS':
        return 52
    elif cnn_type == 'RESNET50':
        return 53
    elif cnn_type == 'INCEPTION_V3':
        return 54
    elif cnn_type == 'VGG16_KERAS':
        return 55
    elif cnn_type == 'VGG19':
        return 56
    elif cnn_type == 'SQUEEZE_NET':
        return 66
    elif cnn_type == 'DENSENET_161':
        return 71
    elif cnn_type == 'DENSENET_121':
        return 72
    else:
        print('Error Unknown CNN Type for random state!!')
        exit()
    return 0