updated args
This commit is contained in:
parent
c45a329df4
commit
d8cb739ab2
|
@ -190,7 +190,7 @@ if __name__ == '__main__':
|
|||
gpu_ids = hyperparams.gpus.split(';')
|
||||
if hyperparams.interactive:
|
||||
# run on 1 gpu
|
||||
print(f'RUNNING INTERACTIVE MODE ON GPUS. gpu ids: {gpu_ids}')
|
||||
print(f'RUNNING INTERACTIVE MODE ON GPUS. gpu ids: {str(gpu_ids)}')
|
||||
os.environ["CUDA_VISIBLE_DEVICES"] = gpu_ids
|
||||
main(hyperparams, None, None)
|
||||
|
||||
|
|
Loading…
Reference in New Issue