Echo/fcb_photo_review
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

调整GPU配置

Browse Source
This commit is contained in:
liuyebo
2024-06-21 14:25:18 +08:00
parent 9584d1fe35
commit 2535b0bf65
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
config/photo_review.py
View File

@@ -17,4 +17,4 @@ IE_BATCH_SIZE = 4
LAYOUT_ANALYSIS = False
# 可使用的GPU
CUDA_VISIBLE_DEVICES = "0,1"
CUDA_VISIBLE_DEVICES = "1"