train_autodeeplab.py: error: unrecognized arguments: --gpu_ids 0,1,2,3

MenghaoGuo / AutoDeeplab

Pytorch Implementation the paper Auto-DeepLab Hierarchical Neural Architecture Search for Semantic Image Segmentation

https://arxiv.org/abs/1901.02985

410 stars 97 forks source link

train_autodeeplab.py: error: unrecognized arguments: --gpu_ids 0,1,2,3 #57

Open NanWangAC opened 4 years ago

NanWangAC commented 4 years ago

hi,could i ask what's wrong with this?

FDU-WeiXiZhang commented 4 years ago

Replace the commas with blank spaces

sorrowyn commented 4 years ago

in train_autodeeplab.py replace
parser.add_argument('--gpu-ids', nargs='', type=int, default=0, help='which GPU to train on (default: 0)') with parser.add_argument('--gpu_ids', nargs='', type=str, default='0', help='which GPU to train on (default: 0)')

--gpus_ids type=str default='0'