search

compbiomed-unito / survhive

Convenient, opinionated wrapper around some (deep) survival models
MIT License
2 stars 0 forks source link

survtrace wrapper does not implement the device control #28

Open IvanUniTO opened 8 months ago

IvanUniTO commented 8 months ago

setting device='cuda:1' has no effect and the code self select the first gpu.

IvanUniTO commented 7 months ago

Fix does not work. probably necessary to modify survtrace code