You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hi there - I just wanted to flag that I ran into an issue when my model was wrapped around a nn.DataParallel - thop gave an error that said the model was on a different device (cpu) to the input (gpu)
The text was updated successfully, but these errors were encountered:
Hi there - I just wanted to flag that I ran into an issue when my model was wrapped around a nn.DataParallel - thop gave an error that said the model was on a different device (cpu) to the input (gpu)
The text was updated successfully, but these errors were encountered: