Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Fix pytorch import issue created in #986
There is an issue with the RPi version after merging of #986. The
pytorch
imports are not moved into the corresponding classes / functions, such now pytorch becomes a dependency when running code, like loading a model. This is non-wanted behaviour, because on RPi we don't install pytorch by default, and causes themanage.py drive --model
to fail. This fix moves the imports into the relevant classes/functions and fixes the issue.