That was what you said. Rewriting output = model (s.float ()) resolved this error.
However, I got the following error. Where does the 500x1 in this error point? I am very sorry to ask you many questions.
RuntimeError: size mismatch, m1: [500 x 1], m2: [40 x 6] at ..\aten\src\TH/generic/THTensorMath.cpp:41