NVIDIA / semantic-segmentation

Nvidia Semantic Segmentation monorepo
BSD 3-Clause "New" or "Revised" License
1.76k stars 388 forks source link

NameError: name 'dataset' is not defined #99

Closed SupriyaB1 closed 3 years ago

SupriyaB1 commented 3 years ago

Hi, While running Inference command - python3 scripts/eval_cityscapes.yml -i Getting error - NameError: name 'dataset' is not defined

Traceback (most recent call last): File "scripts/eval_cityscapes.yml", line 7, in dataset: Cityscapes, NameError: name 'dataset' is not defined

Please help me to solve this error Thanks

SupriyaB1 commented 3 years ago

Solved the issue by running command - python3 -m torch.distributed.launch --nproc_per_node=8 HRNet/semantic-segmentation/train.py --dataset cityscapes --cv 0 --syncbn true --apex true --fp16 true --bs_val 2 --eval val --snapshot "ASSETS_PATH/seg_weights/cityscapes_ocrnet.HRNet_Mscale_outstanding-turtle.pth" --arch ocrnet.HRNet_Mscale --result_dir ./save