BlinkDL / RWKV-LM

RWKV is an RNN with transformer-level LLM performance. It can be directly trained like a GPT (parallelizable). So it's combining the best of RNN and transformer - great performance, fast inference, saves VRAM, fast training, "infinite" ctx_len, and free sentence embedding.
Apache License 2.0
12.32k stars 838 forks source link

如何给RWKV添加其他的神经网络 #93

Closed dataangel closed 1 year ago

dataangel commented 1 year ago

比如给他加一个GCN网络

BlinkDL commented 1 year ago

可以将RWKV看成一个模块,替换原有网络的MHA模块。