Skip to content
This repository was archived by the owner on Oct 31, 2023. It is now read-only.

Make cityscapes binarymask config consistent to polygon config#1214

Open
krumo wants to merge 1 commit intofacebookresearch:mainfrom
krumo:master
Open

Make cityscapes binarymask config consistent to polygon config#1214
krumo wants to merge 1 commit intofacebookresearch:mainfrom
krumo:master

Conversation

@krumo
Copy link

@krumo krumo commented Feb 11, 2020

Hi, I notice that maskrcnn will throw error when running with Cityscapes' binarymask config.

/pytorch/aten/src/THCUNN/ClassNLLCriterion.cu:106: void cunn_ClassNLLCriterion_updateOutput_kernel(Dtype *, Dtype *, Dtype *, long *, Dtype *, int, int, int, int, long) [with Dtype = float, Acctype = float]: block: [0,0,0], thread: [31,0,0] Assertion t >= 0 && t < n_classes failed.

RuntimeError: copy_if failed to synchronize: cudaErrorAssert: device-side assert triggered

However, maskrcnn_benchmark could run with the polygon config succefully. I think the reason is that to reuse the Cityscapes evaluation toolkit, we have to include 'caravan' and 'trailer' in the training classes. Thus, to make maskrcnn could run smoothly with Cityscapes binarymask config, I change the NUM_CLASSES from 9 to 11 and make the experiment setting consistent to the polygon config.

With the changes in the commit, I could train maskrcnn with binarymask config successfully now.

@facebook-github-bot facebook-github-bot added the CLA Signed Do not delete this pull request or issue due to inactivity. label Feb 11, 2020
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

CLA Signed Do not delete this pull request or issue due to inactivity.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants