RuntimeError: Expected object of scalar type Float but got scalar type Double for argument #2 'mat1' in call to _th_addmm

I’m getting this error so can someone help me out with finding the solution

https://colab.research.google.com/drive/1sgrtmjoupP5X-K9wLhs-cVCrz17yqG9r#scrollTo=SW_PWGTmdJ3v


These two seems to be the discrepancies, please correct them up. And please do share the screenshot of the error with your notebook.