
Latest_Query_Transformer_RNN_PyTorch: 在PyTorch中实现论文用于知识追踪的最新查询变换器...
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目在PyTorch框架下实现了论文《用于知识追踪的最新查询变换器RNN》中的模型,旨在提供一个高效、可扩展的知识追踪解决方案。
Last-Query-Transformer-RNN 模型的独特之处在于它仅使用最后一个输入作为变压器编码器中的查询,而不是整个序列。这使得QK矩阵乘法的时间复杂度从O(L^2)降低到O(L),从而允许模型处理更长的序列。
该模型架构的具体用法如下:
```python
from last_query_model import *
seq_len = 100
total_ex = 1200
total_cat = 234
total_in = 2
in_ex, in_cat, in_in = random_data(64, seq_len, total_ex, total_cat, total_in)
model = last_query_model(dim_model=128, heads_en=1)
```
全部评论 (0)
还没有任何评论哟~


