hytseng0509 / CrossDomainFewShot

Cross-Domain Few-Shot Classification via Learned Feature-Wise Transformation (ICLR 2020 spotlight)
323 stars 62 forks source link

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

Open Almostlover0608 opened 3 years ago

Almostlover0608 commented 3 years ago

Is there something wrong about the version of torch or cuda?