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-19 16:14:56 +08:00
parent d2c5899122
commit cd81be7378
3 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
config/photo_review.py
View File

@@ -15,3 +15,6 @@ IE_BATCH_SIZE = 4
# 是否开启布局分析
LAYOUT_ANALYSIS = False
# 可使用的GPU
CUDA_VISIBLE_DEVICES = "1"