Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

what does target actually look like in criterion(out, target)? #580

Open
leejunu opened this issue Jun 3, 2022 · 3 comments
Open

what does target actually look like in criterion(out, target)? #580

leejunu opened this issue Jun 3, 2022 · 3 comments

Comments

@leejunu
Copy link

leejunu commented Jun 3, 2022

Hi, I've been working on a semi-supervised implementation and wanted to write pseudo-label code for this codebase. I thought the naive way of doing criterion(out, model(out)) would have sufficed, but it seems that the target are in some ground-truth format rather than the output of the model. Same issue happens if model has been built using the 'test' parameter in build_ssd as well, this time there is a shape mismatch. Can anyone shed some light on what targets looks like (from the dataloader)

@AuroraRogers
Copy link

Hi,i'm having the same issue here.

@AuroraRogers
Copy link

do you also have the error of "IndexError: The shape of the mask [16, 8732] at index 0 does not match the shape of the indexed tensor [139712, 1] at index 0"

@makun-coder
Copy link

Hello, have you solved it? I have the same problem

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants