home transformers fast_weights

View code on Github

快速称重变压器

这是 PyTorch 中《线性变压器秘密地是快速加权内存系统》一文的带注释的实现。

这是带注释的实现。以下是用于在 Tiny Shakespeare 数据集上训练快速权重转换器的训练代码和一本笔记本。