support not specifying number of channels in ModelMultiHead

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

support not specifying number of channels in ModelMultiHead #244

Closed alex-golts closed 1 year ago

mosheraboh commented 1 year ago

@alex-golts. this one doesn't wait for me, right?

alex-golts commented 1 year ago

@alex-golts. this one doesn't wait for me, right?

I think there was an issue with the CICD that prevented me from seeing where it fails, but now it looks like I can see what causes the error, so I will take a look