Properly set context for detector network (#1670)

dmlc · Jun 9, 2021 · ba36d1b · ba36d1b
1 parent d4bb9a1
commit ba36d1b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gluoncv/model_zoo/smot/general_detector.py b/gluoncv/model_zoo/smot/general_detector.py
@@ -88,7 +88,7 @@ def __init__(self, gpu_id,
         self.ctx = mx.gpu(gpu_id)
 
         self.net = get_net(classes=COCODetection.CLASSES,
-                           ctx=mx.gpu(0),
+                           ctx=self.ctx,
                            model_name=model_name,
                            use_pretrained=use_pretrained,
                            param_path=param_path)