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.

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

当菜鸟(我)使用高版本torch时,遇到的需要注意的地方 Problems I have met when I using high version torch #6

Open
InnerPeace-code opened this issue Jul 16, 2024 · 0 comments

Comments

@InnerPeace-code
Copy link

  1. TypeError: lstm() received an invalid combination of arguments:
    https://blog.csdn.net/yhoon0101/article/details/131649166
    修改代码为:(line 37 in train.py)
    parser.add_argument('--bidirectional', default=True, type=bool, help='Whether use bidirectional LSTM')

  2. RuntimeError: ‘lengths‘ argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor:
    https://blog.csdn.net/weixin_45084253/article/details/122269699
    修改代码为:(line 179 in solver.py)
    estimate_source = self.model(padded_mixture, mixture_lengths.to("cpu"))

我升级为高版本torch是因为遇到runtimeerror: cudnn error: cudnn_status_mapping_error错误,我的原因是因为torch版本和cuda版本不太兼容

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant