You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
RandomPlanckianJitter has a bug from get_planckian_coeffs(mode) that always puts self.pl on gpu 0. self.pl is not registered in the module and does not get moved to the correct gpu.
Moving self.pl from gpu0 to another gpu directly results in incorrect values.
I have created a PR but am not quite sure if registering it as a buffer is how it is normally done in kornia.
At least this fixes the issue when the augmentation is moved to the same gpu.
Describe the bug
RandomPlanckianJitter
has a bug fromget_planckian_coeffs(mode)
that always putsself.pl
on gpu 0.self.pl
is not registered in the module and does not get moved to the correct gpu.Moving
self.pl
from gpu0 to another gpu directly results in incorrect values.Reproduction steps
Expected behavior
Either self.pl should be registered so it moves when the module moves to the correct gpu or self.pl should remain on cpu and get moved in the call.
Environment
This does not occur on all machines.
Happens on this machine with 3070s but not on one with A5000.
Additional context
No response
The text was updated successfully, but these errors were encountered: