maxmustermann/DeepLearningExamples
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Update train.py

Browse source 
Fix `flags.gpu is not None` path
This commit is contained in:
Pablo Ribalta Lorenzo 2021-06-04 12:35:07 +02:00 committed by GitHub
parent ff6f7c6532
commit 3d91dc9385
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
TensorFlow2/Segmentation/MaskRCNN/scripts/train.py
View file

@ -82,7 +82,7 @@ if __name__ == '__main__':
cmd_train += ' ' + ' '.join(remainder)
cmd_eval += ' ' + ' '.join(remainder)
if flags.gpus is not None:
cmd_train = f'CUDA_VISIBLE_DEVICES={",".join(map(str, range(flags.gpus)))} ' + cmd
cmd_train = f'CUDA_VISIBLE_DEVICES={",".join(map(str, range(flags.gpus)))} ' + cmd_train
# print command
line = '-' * shutil.get_terminal_size()[0]