Closed tanujjain closed 2 years ago
Sets the CNN to use gpu for encoding generation if one is available on the machine, else selects cpu (which is the default).
CNN encoding generation can leverage faster computation on gpu.
Tests run successfully on both cpu and gpu machines.
Closing since #179 already introduced the changes.
What
Sets the CNN to use gpu for encoding generation if one is available on the machine, else selects cpu (which is the default).
Why
CNN encoding generation can leverage faster computation on gpu.
Tests
Tests run successfully on both cpu and gpu machines.