New issue

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

#

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

Already on GitHub? # to your account

Jump to bottom

Removes unnessary flatten #50

Merged

jkhenning merged 1 commit into allegroai:main from amirhmk:fix-example-shape-issue

Apr 11, 2023

Contributor

amirhmk commented Apr 5, 2023

This PR fixes #48 by removing the unnessary flatten() that was causing the following error in examples/pytorch and examples/keras:

{"detail":"Error processing request: <AioRpcError of RPC that terminated with:\n\tstatus = StatusCode.INVALID_ARGUMENT\n\tdetails = \"unexpected shape for input 'INPUT__0' for model 'test_model_pytorch'. Expected [1,28,28], got [1,784]\"\n\tdebug_error_string = \"{\"created\":\"@1680665060.017267956\",\"description\":\"Error received from peer ipv4:172.23.0.5:8001\"

This simply makes the input adhere to the shape defined in the tutorial, and one that the model also accepts.


          Removes unnessary flatten

c844a13

Member

jkhenning commented Apr 10, 2023

Hi @amirhmk , sorry for the delay, we'll take a look at this 👍

jkhenning merged commit 41df4eb into allegroai:main

amirhmk deleted the fix-example-shape-issue branch

April 12, 2023 06:01

# for free to join this conversation on GitHub. Already have an account? # to comment

Labels

None yet