/Minimal-PKM-layer

Minimal working example of Product-Key Memory layers

Primary LanguagePythonMIT LicenseMIT

Minimal examples for PKM layer training

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