@Yoga_Hu If you are asking about the “input and target batch don’t match error”, I have used MSE loss instead of NLL loss. If you are asking about the error: “RuntimeError: expected dtype Float but got dtype Long”, I type casted the model outputs to float using .float(). This helped me in solving the problem.