尚未上传
目录 | 说明 |
---|---|
data | Plato-memory数据集,其中train.txt和eval.txt是数据集主体 |
experiment_result | 实验生成的各种记录和数据 |
experiment.py | 实验代码 |
model.py | 模型的实现类 |
plato_memory_dataset.py | 数据集构建代码 |
plato-mini_train.py | 以plato-mini模型为基础的训练代码 |
roformer_train.py | 以roformer模型为基础的训练代码 |
sformer_train.py | 以sformer模型为基础的训练代码 |
utils.py | 预训练需要用到的工具类和函数 |
演示视频.mp4 | SFormer模型的实机效果演示 |
- SFormer,链接:https://github.com/Wu-Jiayang/SFormer
- 链接失效可联系651475459@qq.com