histocartography
histocartography copied to clipboard
how to change the device id?
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?