diff --git a/recognition/arcface_onnx.py b/recognition/arcface_onnx.py
index e77b421..5cda3ec 100644
--- a/recognition/arcface_onnx.py
+++ b/recognition/arcface_onnx.py
@@ -6,7 +6,7 @@
 
 import numpy as np
 import cv2
-import torch
+import torch # hack to use onnxruntime with cuda
 import onnx
 import onnxruntime
 import face_align
diff --git a/recognition/main.py b/recognition/main.py
index b353f68..bddf777 100644
--- a/recognition/main.py
+++ b/recognition/main.py
@@ -5,7 +5,7 @@ import os.path as osp
 import argparse
 import cv2
 import numpy as np
-import torch
+import torch # hack to use onnxruntime with cuda
 import onnxruntime
 from scrfd import SCRFD
 from arcface_onnx import ArcFaceONNX
diff --git a/recognition/scrfd.py b/recognition/scrfd.py
index 2cad754..2a1a156 100644
--- a/recognition/scrfd.py
+++ b/recognition/scrfd.py
@@ -2,7 +2,7 @@
 from __future__ import division
 import datetime
 import numpy as np
-import torch
+import torch # hack to use onnxruntime with cuda
 import onnxruntime
 import os
 import os.path as osp
diff --git a/refacer.py b/refacer.py
index 409c378..97795cb 100644
--- a/refacer.py
+++ b/refacer.py
@@ -1,5 +1,5 @@
 import cv2
-import torch
+import torch # hack to use onnxruntime with cuda
 import onnxruntime as rt
 import sys
 from insightface.app import FaceAnalysis