BiomedSciAI / histocartography

A standardized Python API with necessary preprocessing, machine learning and explainability tools to facilitate graph-analytics in computational pathology.
GNU Affero General Public License v3.0
235 stars 55 forks source link

how to change the device id? #60

Open Extord1108 opened 3 months ago

Extord1108 commented 3 months ago

Hi, thanks for your project, but I have a problem while using it. It seems that you set the "cuda:0" as the default device in some modules such as 'DeepFeatureExtractor', but I have come across some cases that need to change it, could you please add a parameter to allow the selection of the device ID?