Input and hidden tensors are not the same dtype, found input tensor with Double and hidden tensor with Float
(* ̄︶ ̄)