Hello Kamil,
Thank you for the code.The error went away.But I get a runtime error which is as follows:
RuntimeError: expected type torch.FloatTensor but got torch.cuda.FloatTensor
Please assist me in solving this.
Best Regards,
Vijay