Fixing image formatting for test data sets #2104
Closed
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Description
A number of utility functions are defined in conftest.py for loading default subsets of data (e.g. cifar10 and mnist) which are subsequently used in tests. Some of these methods, depending on the framework used (e.g. PyTorch), change the format of the images from HxWxC to CxHxW, and do so by calling np.reshape, which incorrectly changes the format of the images.
To fix, transpose should be used in place of reshape. This subsequently requires updating any tests which used the incorrect format for images.
Fixes #2101
Type of change
Please check all relevant options.
Testing
Please describe the tests that you ran to verify your changes. Consider listing any relevant details of your test configuration.
Test Configuration:
Checklist