Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

GPU训练 #24

Open
18813055625 opened this issue Jul 26, 2018 · 3 comments
Open

GPU训练 #24

18813055625 opened this issue Jul 26, 2018 · 3 comments

Comments

@18813055625
Copy link

crnn训练时，我用Keras，在CPU上可以运行，用GPU运行时，会报错。不知道有没有遇到同样问题的小伙伴，可以加qq 2541899447一起讨论。报错原因：ValueError: Shape must be rank 1 but is rank 4 for 'batch_normalization_1/cond/FusedBatchNorm' (op: 'FusedBatchNorm') with input shapes: [?,4,?,512], [1,4,1,1], [1,4,1,1], [1,4,1,1], [1,4,1,1]
我估计是batchnormlization出错。
@18813055625
Copy link
Author

是Keras版本的问题，已解决

@Lanme
Copy link

Lanme commented Jun 17, 2019

是Keras版本的问题，已解决

请问你用了什么版本？我也是这个错误。

@Lanme
Copy link

Lanme commented Jun 17, 2019 •
edited

解决了，我的是keras2.2.4，keras-team/keras@e3a2f7d

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@Lanme @18813055625