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

TinyImageNetDataSetIterator doesn't validate or correctly use input shapes #5234

AlexDBlack opened this issue May 18, 2018 · 1 comment


Copy link

commented May 18, 2018

The int[] imgDim arg: I had assumed 3,224,224 was valid input. But it's not: it should have been 224,224.
I only realized this much later once I got an obscure shape error during feed forward.

Edit: Even that's not enough... this simple test fails - the shapes are wrong (outputting 64x64 not 224x224 images)

        DataSetIterator iter = new TinyImageNetDataSetIterator(1, new int[]{224,224}, DataSetType.TRAIN, null, 12345);
        DataSet ds =;

        assertArrayEquals(new int[]{1,3,224,224}, ds.getFeatures().shape());
        assertArrayEquals(new int[]{1,200}, ds.getLabels().shape());

@AlexDBlack AlexDBlack changed the title TinyImageNetDataSetIterator doesn't validate input shapes TinyImageNetDataSetIterator doesn't validate or correctly use input shapes May 18, 2018

@AlexDBlack AlexDBlack added this to the DL4J/Arbiter/DataVec Planning Sprint (01 June) milestone May 18, 2018

AlexDBlack added a commit that referenced this issue May 21, 2018

This comment has been minimized.

Copy link

commented Sep 22, 2018

This thread has been automatically locked since there has not been any recent activity after it was closed. Please open a new issue for related bugs.

@lock lock bot locked and limited conversation to collaborators Sep 22, 2018

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
None yet
1 participant
You can’t perform that action at this time.