8000 GitHub - SKY-GOOD/convit: This repo is to implement convit using MindSpore
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

SKY-GOOD/convit

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 

Repository files navigation

convit

This repo is to implement convit using MindSpore

Finish

convit.py -> 能够跑通main测试函数,推理结果与pytorch结果一致

TO DO

1、model.train 方式训练

2、pynative+混合编程训练

3、差异文档分析编写

更新日志

2022/10/9:

使用pthtockpt.py脚本,将pth模型转换为ckpt模型,并进行了精度推理测试

top1 acc: 0.72576
top5 acc: 0.9146

略低于论文中的精度,但认为可以通过调整训练参数

2022/10/8:

MLP层中,将

self.act = nn.GELU()

修改为

self.act = nn.GELU(approximate=False)

减小了与pytorch的误差

About

This repo is to implement convit using MindSpore

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%
0