In this simple commit I just fixed a couple of issues of the notebook. Moreover I changed the default device from cuda to cpu (since this is a simple example I think that all the people, even who does not own a graphics card have to be able to try it out of the box). Moreover I removed a useless empty code section
Type of change
Please delete options that are not relevant.
[x] Bug fix (non-breaking change which fixes an issue)
[ ] New feature (non-breaking change which adds functionality)
[ ] Breaking change (fix or feature that would cause existing functionality to not work as expected)
[ ] This change requires a documentation update
Checklist:
[x] The code follows the Python style guide.
[x] Code and files are well organized.
[ ] All tests pass.
[ ] New code is covered by tests.
[x] [Optional] We would be very happy if gitmoji :technologist: could be used to assist the commit message :speech_balloon:!
Description
In this simple commit I just fixed a couple of issues of the notebook. Moreover I changed the default device from
cuda
tocpu
(since this is a simple example I think that all the people, even who does not own a graphics card have to be able to try it out of the box). Moreover I removed a useless empty code sectionType of change
Please delete options that are not relevant.
Checklist: