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

Make cityscapes binarymask config consistent to polygon config #1214

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

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