Closed 1Maxnet1 closed 1 year ago
When a model was trained with CUDA, one needs to enable a mapping parameter to evaluate it on CPU. (and the other way). This adds some code to automatically detect the used device and mapps the model accordingly.
Closes #15
When a model was trained with CUDA, one needs to enable a mapping parameter to evaluate it on CPU. (and the other way). This adds some code to automatically detect the used device and mapps the model accordingly.
Closes #15