Performance issues?

Flux is slower than PyTorch in CPU mode, Conv2d is even 2x slower (issue), but in GPU mode they’re comparable.

3 Likes