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
RuntimeError: size mismatch, m1: [1 x 588], m2: [12 x 10] at /pytorch/aten/src/THC/generic/THCTensorMathBlas.cu:268 #12
Comments
Hi @vainaijr, I would need more information in order to diagnose: e.g. the model architecture, how you are instantiating the Please also provide any other information you think would be useful. Many thanks, |
hello,
I pass images from top neural network to second one, then to middle, then to bottom, and then get 10 probabilities, which I pass to the top. I have to use flashtorch to visualize what each neural network learns, this is different from .deep learning where we have only one model, here I use multiple encoders, decoders, and pass output top to bottom, or bottom to top. |
Thanks @vainaijr,
For certain types of architecture, especially if the liner layers are interwoven and hence can't be separated, you might have to set the You can do so by passing it in on the object instantiation or by reassigning the attribute. I.e.
Or
Let me know how it goes. Many thanks, |
That's great @vainaijr, looking forward to hearing what insights you gain with FlashTorch. |
hello,
does flashtorch work on ensemble model, I passed output of one neural network to another, and got error.
I used
g_ascent.visualize(model.encoder[0], title='conv');
The text was updated successfully, but these errors were encountered: