From ba36d1b8448dda5edc4ccf81e771ed3dbbd9fd41 Mon Sep 17 00:00:00 2001
From: Mark Steve Samson <26679+marksteve@users.noreply.github.com>
Date: Wed, 9 Jun 2021 13:00:43 +0800
Subject: [PATCH] Properly set context for detector network (#1670)

---
 gluoncv/model_zoo/smot/general_detector.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gluoncv/model_zoo/smot/general_detector.py b/gluoncv/model_zoo/smot/general_detector.py
index 66b8c14cec..77164617e6 100644
--- 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)