Ascend910训练CSWin-Transformer失败

最新推荐文章于 2024-08-10 21:26:27 发布

xiao | yang

最新推荐文章于 2024-08-10 21:26:27 发布

阅读量411

点赞数

分类专栏：深度学习文章标签： transformer 深度学习人工智能

本文链接：https://blog.csdn.net/xi_xiyu/article/details/127645715

版权

使用ModelArts平台，基于tensorflow1.15-mindspore1.5.1-cann5.0.2-euler2.8-aarch64镜像进行CSWin-Transformer训练时遇到RuntimeError。错误信息提示dy_w*stride_w的值必须在[2,4096]范围内，实际为1。问题可能源于前向传播的某层。建议通过MindSpore的Print操作进行日志排查。" 123669425,11573455,使用antd和react获取接口数据实现Select组件,"['前端框架', 'react', '前端']

摘要由CSDN通过智能技术生成

源码：https://gitee.com/lljyoyo1995/cswin.git

【操作步骤&问题现象】

1、train.py拉起训练失败

【截图信息】

运行平台：在ModelArts
镜像：tensorflow1.15-mindspore1.5.1-cann5.0.2-euler2.8-aarch64

RuntimeError: ({'errCode': 'E60011', 'range': '[2,4096]', 'attr_name': 'dy_w*stride_w', 'value': 1}, 'In op, the [dy_w*stride_w] must in range [[2,4096]], actual is [1]')

【日志信息】（可选，上传日志内容或者附件）

Traceback (most recent call last):
File "/home/ma-user/modelarts/user-job-dir/cswin/train.py", line 101, in
main()
File "/home/ma-user/modelarts/user-job-dir/cswin/train.py", line 92, in main
dataset_sink_mode = False)
File "/home/ma-user/anaconda/lib/python3.7/site-packages/mindspore/train/model.py", line 726, in