Runtime Error in Pyro Model Training


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