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
137 stars 34 forks source link

optimized train time for a use case of small samples and large batch #268

Closed mosheraboh closed 1 year ago

mosheraboh commented 1 year ago

Optimize nicely the train running time in such a use case (small samples and large batch size) Leaving ndict optimization for the future