Andrea

Results 3 comments of Andrea

I picked up this issue yesterday. I am toying around with it, based on @xaviergonz and @another-guy previous comments, for now but if people are still interested I can try...

After trying around a bit more it might be an issue with how `Lux.apply` expects `Matrix` rather than `Vector` when a `BatchNorm` layer is in the chain. This other MWE...