RuntimeError: Expected object of scalar type Byte but got scalar type Float for argument #2 'mat2'

1

Please help me with this. Can’t trace the error. Check comments for entire code.

error
error817×117 10.5 KB

2

error1
error1828×660 47.6 KB

3

error2
error21095×644 57.5 KB

4

I just realized that I was missing the flattening step. It solved the problem. :sweat_smile: