typo fixed

apache · Mar 15, 2019 · abf3c34 · abf3c34
1 parent 4c8007a
commit abf3c34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/topi/python/topi/cuda/ssd/multibox.py b/topi/python/topi/cuda/ssd/multibox.py
@@ -183,7 +183,7 @@ def transform_loc_pre(cls_prob, valid_count, temp_valid_count, temp_cls_id, temp
     max_threads = int(
         tvm.target.current_target(allow_none=False).max_num_threads)
     nthread_tx = max_threads
-    nthread_bx = (batch_size * num_classes * num_anchors) // max_threads + 1
+    nthread_bx = (batch_size *  num_anchors) // max_threads + 1
     tx = tvm.thread_axis("threadIdx.x")
     bx = tvm.thread_axis("blockIdx.x")
     ib.scope_attr(tx, "thread_extent", nthread_tx)