https://keroseneheaterreview.blogspot.com/2024/05/runtimeerror-given-groups1-weight-of.html