search

teamtomo / libtilt

Image processing for cryo-electron microscopy in PyTorch
Other
14 stars 5 forks source link

add soft edge code breaks if device is cuda #57

Open alisterburt opened 8 months ago

alisterburt commented 8 months ago

https://github.com/teamtomo/libtilt/blob/958e58096746a00e437f8714eca11815af4a669e/src/libtilt/shapes/soft_edge.py#L11

The above line breaks if the device is cuda, needs fixing

pointed out by @rsanchezgarc