[Fix] rotation classifier input move to model device

[felixdittrich92]

This PR:

• fix moving rotation classifier input to cuda device if given

Closes:
#1020

[codecov]

Codecov Report

Merging #1039 (29f67df) into main (75aa42a) will decrease coverage by 0.01%.
The diff coverage is 100.00%.

@@            Coverage Diff             @@
##             main    #1039      +/-   ##
==========================================
- Coverage   94.93%   94.91%   -0.02%     
==========================================
  Files         135      135              
  Lines        5625     5626       +1     
==========================================
  Hits         5340     5340              
- Misses        285      286       +1     

Flag	Coverage Δ
unittests	94.91% <100.00%> (-0.02%)	⬇️

Flags with carried forward coverage won't be shown. Click here to find out more.

Impacted Files	Coverage Δ
doctr/models/classification/predictor/pytorch.py	95.00% <100.00%> (+0.26%)	⬆️
doctr/transforms/functional/base.py	95.65% <0.00%> (-1.45%)	⬇️
doctr/transforms/modules/base.py	94.59% <0.00%> (ø)

Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here.

[charlesmindee]

thanks!

[minouei-kl]

for me this fix didn't worked I had to change it like this:

        _device = next(self.model.parameters()).device
        predicted_batches = [self.model(batch.to(device=_device)).to(device=_device) for batch in processed_batches]

[odulcy-mindee]

@minouei-kl thanks! @felixdittrich92 saw this problem in #1020 but batch has not been moved to the good device. I'll fix that.