forked from XiuzeZhou/NASA
-
Notifications
You must be signed in to change notification settings - Fork 1
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
设备不一致 #1
Comments
终于是复现成功了 命令行日志
报错堆栈
|
谢谢大佬,非常感谢o(╥﹏╥)o |
要是解决了,可以考虑 Close Issue 。 Edit: 或者我做个 Pull Request 作为 Closer 吧 |
Dragon1573
added a commit
to Dragon1573/NASA
that referenced
this issue
Nov 22, 2024
- Several tensors and modules forget to move to target device. - They left on CPU while user running the notebook with a CUDA version of PyTorch, hence a "device mismatch" error occur. Signed-off-by: Dragon1573 <49941141+Dragon1573@users.noreply.github.com>
Closed
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
@Dragon1573
The text was updated successfully, but these errors were encountered: