From 7f679cb51be6c5f1401df12e833102deee3ab580 Mon Sep 17 00:00:00 2001 From: panjiacheng Date: Tue, 6 Nov 2018 10:45:30 -0800 Subject: [PATCH] Prediction: minor changes. --- modules/tools/prediction/mlp_train/cruiseMLP_train.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/modules/tools/prediction/mlp_train/cruiseMLP_train.py b/modules/tools/prediction/mlp_train/cruiseMLP_train.py index cf4c6ee82e..d38fdfc4de 100644 --- a/modules/tools/prediction/mlp_train/cruiseMLP_train.py +++ b/modules/tools/prediction/mlp_train/cruiseMLP_train.py @@ -517,13 +517,13 @@ if __name__ == "__main__": parser.add_argument('valid_file', type=str, help='validation data (h5)') parser.add_argument('-d', '--data_loader', action='store_true', \ help='Use the dataloader (when memory size is smaller than dataset size)') - parser.add_argument('-g', '--gpu_num', type=int, default=0, \ - help='Specify which GPU to use.') + #parser.add_argument('-g', '--gpu_num', type=int, default=0, \ + # help='Specify which GPU to use.') args = parser.parse_args() - os.environ['CUDA_DEVICE_ORDER'] = 'PCI_BUS_ID' #specifies the same order as nvidia-smi - os.environ['CUDA_VISIBLE_DEVICES'] = str(args.gpu_num) + #os.environ['CUDA_DEVICE_ORDER'] = 'PCI_BUS_ID' #specifies the same order as nvidia-smi + #os.environ['CUDA_VISIBLE_DEVICES'] = str(args.gpu_num) if not args.data_loader: -- GitLab