Skip to content

Wu-Jiayang/Plato-memory

Repository files navigation

Plato-memory


论文地址

尚未上传

结构说明

目录 说明
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模型的实机效果演示

相关资源

About

Plato-memory

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages