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...
I will have a stab at it next week!
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...