Skip to content

move the relevant improved_model to device instead of the model from previous section#1

Open
saied89 wants to merge 2 commits intoguyernest:mainfrom
saied89:main
Open

move the relevant improved_model to device instead of the model from previous section#1
saied89 wants to merge 2 commits intoguyernest:mainfrom
saied89:main

Conversation

@saied89
Copy link

@saied89 saied89 commented Apr 15, 2025

This causes an error when cuda is available since the actual model in the training isn't on the cuda device.

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

Successfully merging this pull request may close these issues.

1 participant