search

rotemtzaban / STIT

MIT License
1.2k stars 170 forks source link

fix bug of type dismatch. #16

Closed mazzzystar closed 2 years ago

mazzzystar commented 2 years ago

First convert Tensor to PIL for Resize, then covert back to Tensor for .to() operation.