-
Notifications
You must be signed in to change notification settings - Fork 653
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
Possibility to ignore the reconstruction/decoder? #81
Comments
Hi @fastesol if you want to get rid of decoder/reconstruction, you should also exclude the loss function it is using which is 'mse' |
Thank you @JoyJulianGomes can you instruct the required changes in the code please? Sorry I’m new to this field. |
@fastesol as you are new to this, working out the problem yourself would be better for the long run than just telling you what to change. I can give you some hints but you have to do it yourself.
P.S.: I am new to this as well |
@muaaz-khalid You can just set the parameter |
I think, reconstruction/decoder is not much helpful in learning the model, but creating an extra computation overhead. So, in order to prove this hypothesis/thought (or whatever you call it), is it possible to remove the decoder part?
This is what I tried so far.
but getting the following error
I'm sure, I'm doing it wrong way, probably have to reshape the out_caps to something else. Please help.
Thank you!
The text was updated successfully, but these errors were encountered: