Python draw_img_classifier_to_file примеры использования

Пример #1

def test_png_generation():
    DATASET = "cifar10"
    ARCH = "resnet20_cifar"
    model, zeros_mask_dict = common.setup_test(ARCH, DATASET, parallel=True)
    # 2 different ways to create a PNG
    distiller.draw_img_classifier_to_file(model, 'model.png', DATASET, True)
    distiller.draw_img_classifier_to_file(model, 'model.png', DATASET, False)

Пример #2

Файл: test_model_summary.py Проект: jtdutta1/distiller

def test_png_generation(display_param_nodes):
    dataset = "cifar10"
    arch = "resnet20_cifar"
    model, _ = common.setup_test(arch, dataset, parallel=True)
    # 2 different ways to create a PNG
    distiller.draw_img_classifier_to_file(model, 'model.png', dataset,
                                          display_param_nodes)