issues
search
computational-cell-analytics
/
micro-sam
Segment Anything for Microscopy
https://computational-cell-analytics.github.io/micro-sam/
MIT License
311
stars
35
forks
source link
Remove device input for TrainableSAM
#632
Closed
anwai98
closed
3 weeks ago
anwai98
commented
3 weeks ago
This PR addresses:
Removing
device
input argument for
TrainableSAM
. The device is now fetched automatically from
self.sam
.
Adds
non_blocking=True
for moving objects across devices (speeds up the processes by a bit)
This PR addresses:
deviceinput argument forTrainableSAM. The device is now fetched automatically fromself.sam.non_blocking=Truefor moving objects across devices (speeds up the processes by a bit)