Python plot_line_tradeoff Beispiele

Beispiel #1

    Y_test,
    thresholds,
    batchsize=TEST_BATCHSIZE,
    verbose=True)

g_diffs *= 1000.

# 显示原始网络与带有分支的网络测试精度与运行时间的关系图
visualize.plot_line_tradeoff(g_accs,
                             g_diffs,
                             g_ts,
                             g_exits,
                             g_baseacc,
                             g_basediff,
                             all_samples=False,
                             inc_amt=0.0001,
                             our_label='BranchyLeNet',
                             orig_label='LeNet',
                             xlabel='Runtime(ms)',
                             title='LeNet Gpu',
                             output_path=SAVE_PATH,
                             output_name='lenet_gpu(' +
                             str(TRAIN_NUM_EPOCHES) + ')')
# 将结果保存为csv文件
utils.branchy_save_csv(g_baseacc,
                       g_basediff,
                       g_accs,
                       g_diffs,
                       g_exits,
                       g_ts,
                       filename=CSV_NAME)

Beispiel #2

Datei: experiment_resnet_cifar10.py Projekt: SanityLacking/branchynet

                                                      verbose=True)
# g_ts, g_accs, g_diffs, g_exits = utils.screen_leaky(leakyNet, x_test, y_test, thresholds, inc_amt=-0.1,
#                                                     batchsize=TEST_BATCHSIZE, verbose=True)

#convert to ms
g_diffs *= 1000.

# In[ ]:

visualize.plot_line_tradeoff(g_accs,
                             g_diffs,
                             g_ts,
                             g_exits,
                             g_baseacc,
                             g_basediff,
                             all_samples=False,
                             inc_amt=-0.0001000,
                             our_label='BranchyResNet',
                             orig_label='ResNet',
                             xlabel='Runtime (ms)',
                             title='ResNet GPU',
                             output_path='_figs/resnet_gpu.pdf')

# In[ ]:

#CPU
branchyNet.to_cpu()
c_ts, c_accs, c_diffs, c_exits = utils.screen_branchy(branchyNet,
                                                      x_test,
                                                      y_test,
                                                      thresholds,