updated args

This commit is contained in:
William Falcon 2019-06-25 18:44:50 -04:00
parent c45a329df4
commit d8cb739ab2
1 changed files with 1 additions and 1 deletions

View File

@ -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)