Update testbench

vysarge · Dec 10, 2017 · 094de9c · 094de9c
1 parent 4d18336
commit 094de9c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/os_arch/tb.py b/os_arch/tb.py
@@ -136,8 +136,9 @@ def one_layer(image_size, filter_size, in_chn, out_chn, block_size, ifmap, weigh
         out_chn = [16, 16, 8, None]
         block_size = [8, 8, 8, None]
         num_nonzeros = [[8, 8, 8, None], [4, 4, 4, None], [2, 2, 2, None], [1, 1, 1, None]]
-        #pruner_name = "NaivePruner"
-        pruner_name = "ClusteredPruner"
+        pruner_name = "NaivePruner"
+        print("Outputs may take a long time to start appearing (~5 mins); be patient!")
+        #pruner_name = "ClusteredPruner"
 
         # Load weights and biases from file
         saved_weights = np.load("../cnn/98.npy")