模型训练时候报错，运行train.py(win和linux系统报错不同)

[xiaoxiaoli666]

感谢博主开源！从读研开始一直关注博主，谢谢博主贡献代码。 下边是我运行时候遇到的两个问题，如果有遇到同样问题可以参考。 在运行时候会遇到两个问题，在win系统上，会报错ValueError: setting an array element with a sequence. The requested array has an inhomogeneous shape after 1 dimensions. The detected shape was (6000,) + inhomogeneous part（pytorch版本为2.0.1）.linux系统上会报错RuntimeError: CuDNN error: CUDNN_STATUS-NOT_INITIALIZED（torch==1.2.0）。 查了下一个是数据长度不匹配，一个是cuda加速问题，在linux上cuda=false，用cpu就没这个问题。其他方法也懒得在尝试了 第二个是predict.py文件，如果显示匹配不到qt类似问题，打不开图片，环境装个pyqt5的包即可解决。

[xiaoxiaoli666]

附一下其他的解决办法（cudnn error）：https://blog.csdn.net/weixin_43935696/article/details/114950143

[shawnl111]

第一个问题，pip install numpy==1.21.6 可以解决

[xiaoxiaoli666]

太感谢您了！！！现在知道报错问题了~很难过我现在不做这个方向了，感谢您的开源付出~祝好~

李智慧 @.***

 

------------------ 原始邮件 ------------------ 发件人: "bubbliiiing/arcface-pytorch" @.>; 发送时间: 2024年4月5日(星期五) 下午2:00 @.>; @.@.>; 主题: Re: [bubbliiiing/arcface-pytorch] 模型训练时候报错，运行train.py(win和linux系统报错不同) (Issue #36)

第一个问题，pip install numpy==1.21.6 可以解决

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>