search

BiomedSciAI / fuse-med-ml

A python framework accelerating ML based discovery in the medical field by encouraging code reuse. Batteries included :)
Apache License 2.0
134 stars 34 forks source link

add a flag to metric to avoid multigpu collection #311

Closed mosheraboh closed 1 year ago

mosheraboh commented 1 year ago

LGTM, Thanks!

Three general comments that come to my mind:

  1. Will be interesting to measure the speedup.
  2. In the future it might be cool to support partial multi gpu metric collection. We'll be able to control the tradeoff between the running time and the metric precision.
  3. I guess we should start considering multi-gpu unit testing soon 😃

Thanks! my main usage was to avoid collecting tricky cases.