diff --git a/config/photo_review.py b/config/photo_review.py index c1d2318..bd0e716 100644 --- a/config/photo_review.py +++ b/config/photo_review.py @@ -17,4 +17,4 @@ IE_BATCH_SIZE = 4 LAYOUT_ANALYSIS = False # 可使用的GPU -CUDA_VISIBLE_DEVICES = "1" +CUDA_VISIBLE_DEVICES = "0,1"