Is CPU mode available? #6
Description
I was trying to run some examples but apparently GPU with 8GB of memory is not enough. When I checked the code of the package I noticed that there is a DEVICE variable that is never used and there is hardcoded "cuda" here.
Is there a way to use this package without GPU?