Giters

BiomedSciAI / histocartography

A standardized Python API with necessary preprocessing, machine learning and explainability tools to facilitate graph-analytics in computational pathology.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

how to change the device id?

Extord1108 opened this issue · comments

commented

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?