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

About the multiple categories in code #22

Open
alimagic opened this issue Dec 11, 2020 · 1 comment
Open

About the multiple categories in code #22

alimagic opened this issue Dec 11, 2020 · 1 comment

Comments

@alimagic
Copy link

Dear Author:
When I use this code to train for multiple categories, I find the code in "dataset.py" about line 37 "self._valid_ids = [1]" .Does it need to be modified? For example, two categories, self._valid_ids = [1, 2].

@ZeroE04
Copy link
Owner

ZeroE04 commented Dec 12, 2020

Yes. I forgot to emphasize that. Sorry.

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

2 participants