search

almaan / stereoscope

Spatial mapping of cell types by integration of transcriptomics data
MIT License
87 stars 26 forks source link

gpu utilization #17

Open giovp opened 4 years ago

giovp commented 4 years ago

Hi,

so I happen to have really low gpu utilization with default parameters (~10%), if I increase batch size to 500 I get to 20%, in both cases on a GTX Titan X (which is a pretty bad gpu). Is there any suggestion on how to best optimize trainer params ?

Thank you!