Vojta Molda

I encountered the same issue and the fix suggested by @BradLarson works well. I'd suggest to wrap the it in a `#if canImport(...)` so it doesn't affect backward compatibility for...

Thanks for the quick response! I just tested `3.8.1` and everything works like a charm. Sorry about the confusion with `#if canImport`.

Below is a ResNet-like architecture with about 80k parameters that I put together. It uses only 3x3 convolutions. There's an initial convolution and three ResNet blocks stacked on top of...