RuntimeError: expected scalar type Double but found Float For the data in the model that needs to be re scanned, add a. Float . Long for the label that needs to be scanned