search

ElephantGit / SemanticSegmentationUsingFPN_PanopticFeaturePyramidNetworks

Other
81 stars 21 forks source link

Could not use multiGPU #1

Closed ParanoidW closed 5 years ago

ParanoidW commented 5 years ago

Actually, the current version doesn't support multiGPU in train_val.py.

ElephantGit commented 5 years ago

what the command line do you use to us multiGPUs?

ParanoidW commented 5 years ago

what the command line do you use to us multiGPUs?

python train_val.py --dataset Cityscapes --save_dir /path/to/run --mGPUs True --gpu_ids 0,1,2 Just followed the README. However, the train_val.py doesn't support mGPUs and gpu_ids arguments.

ElephantGit commented 5 years ago

how many gpus do you have? three?

ParanoidW commented 5 years ago

I have 8 in total and have tested on 3.

ElephantGit commented 5 years ago

Emmm, could you copy your error message here?

ParanoidW commented 5 years ago

It tells me that the parser can't parse mGPUs and gpu_ids.

ElephantGit commented 5 years ago

sorry, this is my fault, recent commit cover the mGPUs version code. This version does not support mGPUs training.

ParanoidW commented 5 years ago

Thanks anyway~

ElephantGit commented 5 years ago

I roll-back the recent commit, you can try now to see whether it work now.