Phirefly9 / Minimal-PKM-layer

Minimal working example of Product-Key Memory layers

MIT License

0 stars 0 forks source link

Minimal examples for PKM layer training

contains an implementation of PKM layer training for both pytorch and tensorflow2