Hello
I’ve faced a technical challenge while working with Pyro, and I could use some guidance in it.
Whenever I attempt to run my Pyro model, I’m facing a runtime error that I can’t able to resolve. The error message states ‘RuntimeError: shape ‘[32, 64, 128]’ is invalid for input of size 245760’. Despite reviewing my code and checking for potential issues, I’m still unable to pinpoint the source of the problem.
- I’m using Pyro version 1.7.0.
- The error occurs specifically when I try to train my model using the ‘train()’ function.
- I’ve followed the Pyro documentation and examples closely but haven’t been able to overcome this error. Pyro Documentation — Pyro documentation
If anyone has faced a similar issue or has help on how to debug it effectively, I would greatly appreciate your help .
Thanks in advance for your help!
thank you
stevediaz