RuntimeError: mat1 and mat2 must have the same dtype

[ptrblck]

Your code is still not executable and not properly formatted.
However, it seems you have solved the issue by transforming everything to float32.

@Swetha_Kerahalli

I have changed it to x.to(torch.float32), but after doing this getting error as
RuntimeError: mat1 and mat2 shapes cannot be multiplied (480x8568 and 784x100)

This shape mismatch is raised if the in_features of the linear layer do not match the feature dimension of the input activation, so use in_features=8568 and it should work.