From 73470b28e3a102093c5ef2e8097e32bedff50c74 Mon Sep 17 00:00:00 2001 From: RachelXu7 Date: Tue, 13 Sep 2022 11:56:37 +0000 Subject: [PATCH] Add ACT/FullQuant Demo --- deploy/auto_compression/run.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/deploy/auto_compression/run.py b/deploy/auto_compression/run.py index d9f21833..d20a2f0a 100644 --- a/deploy/auto_compression/run.py +++ b/deploy/auto_compression/run.py @@ -65,9 +65,7 @@ def eval_function(exe, compiled_test_program, test_feed_names, label = np.array(label).reshape((-1, 1)) sort_array = pred.argsort(axis=1) top_1_pred = sort_array[:, -1:][:, ::-1] - # print(label, top_1_pred) top_1 = np.mean(label == top_1_pred) - # print(top_1) top_5_pred = sort_array[:, -5:][:, ::-1] acc_num = 0 for i in range(len(label)): -- GitLab