RuntimeError: Found dtype Long but expected Double

The print statements in the fit method look correct and I don’t know why they wouldn’t show up. Could you add more debug print statements e.g. at the beginning of the fit method and make sure it’s shown?