loss wrapper to batch dict

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

Closed avihu111 closed 1 year ago

avihu111 commented 1 year ago

wraps a loss module to use the batch dict.

avihu111 commented 1 year ago

I added a usage example, hoping for faster code understanding. @akoseki2 would you have another look please?