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

Added TinyImageNet together with example and test. #61

Merged
merged 4 commits into from
May 20, 2020
Merged

Added TinyImageNet together with example and test. #61

merged 4 commits into from
May 20, 2020

Conversation

AndreaCossu
Copy link
Collaborator

  1. Class labels are in the range(0, 199), so the user should use an appropriate output layer or convert labels into a smaller range. We can easily extend the class to do that based on user input.
  2. The class returns numpy array (batch, channels, W, H). I saw that ImageNet leverages pytorch transforms, but since it is the only one that does so, I preferred to be consistent with all others benchmarks.
  3. The class does not shuffle examples.

I can easily fix these two points in the near future but maybe we should better define how we want to manage datasets, as in this issue.

@vlomonaco
Copy link
Member

vlomonaco commented May 20, 2020

Yes, it's good as it is now. Let's wait for #10 to be fixed based on the suggestions of #39

@vlomonaco vlomonaco merged commit 5c5d67e into ContinualAI:master May 20, 2020
@AndreaCossu AndreaCossu linked an issue May 20, 2020 that may be closed by this pull request
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

Successfully merging this pull request may close these issues.

Add Tiny-ImageNet
2 participants