Fixing bug related to compress / decompress on GPU

InterDigitalInc / CompressAI

A PyTorch library and evaluation platform for end-to-end compression research

https://interdigitalinc.github.io/CompressAI/

BSD 3-Clause Clear License

1.19k stars 232 forks source link

Closed navid-mahmoudian closed 3 years ago

navid-mahmoudian commented 3 years ago

I added where it is necessary to emphasize in which device the matrix must be created.

jbegaint commented 3 years ago

Thanks a lot Navid!