Skip to content

devgvh/ning_code

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

20 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

一个Transformer结构的简要代码实现,并做了一个能支持加减乘除四则运算的模型 代码讲解见博客:https://mp.weixin.qq.com/s/a23uVhSzDqQyHWt2QHznEg

两数相加.py:Transformer的代码实现,以及四则运算模型的训练代码

四则运算_推理代码.py 推理代码示例。加载已训练好的模型,直接推理

model_plus_final.pth:已经训练好的,成功率约70%

支持gpu运行 代码基于网上免费教材改写,欢迎免费使用

About

code learning for ning

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages