search

BioMedIA / deepali

Image, point set, and surface registration in PyTorch.
https://biomedia.github.io/deepali/
Apache License 2.0
26 stars 6 forks source link

[spatial] Move transformer buffers to transform.device on init #85

Closed aschuh-hf closed 1 year ago

aschuh-hf commented 1 year ago

Do not require an explicit transformer.to() call when transform was moved to device previously.