FNet-pytorch
FNet-pytorch copied to clipboard
are you ready to Implement embedding part?
trafficstars
thanks for your impl, and I found that you may not code emdedding part, I think the part is important too.
I suggest that you could write a demo from input to prediction if you have free time, excespecially how to code embedding part.
@BinchaoPeng you can checkout our repo at https://github.com/erksch/fnet-pytorch. We implemented the whole architecture and offer official checkpoints converted from the original Jax model.