Update train.py
Fix `flags.gpu is not None` path
This commit is contained in:
parent
ff6f7c6532
commit
3d91dc9385
|
@ -82,7 +82,7 @@ if __name__ == '__main__':
|
||||||
cmd_train += ' ' + ' '.join(remainder)
|
cmd_train += ' ' + ' '.join(remainder)
|
||||||
cmd_eval += ' ' + ' '.join(remainder)
|
cmd_eval += ' ' + ' '.join(remainder)
|
||||||
if flags.gpus is not None:
|
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
|
# print command
|
||||||
line = '-' * shutil.get_terminal_size()[0]
|
line = '-' * shutil.get_terminal_size()[0]
|
||||||
|
|
Loading…
Reference in a new issue