Skip to content

Commit 0a0c165

Browse files
authored
Update MARGE.py
1 parent 54aba26 commit 0a0c165

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

MARGE.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -187,7 +187,7 @@ def MARGE(confile):
187187
"GPU, your system and/or environment may not be" + \
188188
" set up correctly.")
189189
sys.exit(1)
190-
if use_cpu and gpu_detected:
190+
if use_cpu and ngpu_detected:
191191
logger.warning("GPU detected, but configuration file says to use the CPU. " +\
192192
"MARGE execution will be slower compared to using the GPU.\n")
193193
if use_cpu and trainflag:

0 commit comments

Comments
 (0)