Skip to content

Commit e47974b

Browse files
authored
fix the bug in dataloader stop running
1 parent cd5776d commit e47974b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

data/config.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
11
# config.py
22

33
# gets home dir cross platform
4-
4+
import cv2
5+
cv2.setNumThreads(0) # pytorch issue 1355: possible deadlock in dataloader
56
# note: if you used our download scripts, this should be right
67
VOCroot = '/home/user/Database/VOCdevkit' # path to VOCdevkit root dir
78
COCOroot = '/home/user/Database/MSCOCO2017'

0 commit comments

Comments
 (0)