mlm-scoring icon indicating copy to clipboard operation
mlm-scoring copied to clipboard

Added CPU support

Open zolastro opened this issue 1 year ago • 0 comments

Fix #9

CUDA was always used when using torch model, even if CPU was selected.

Changes:

  • Use self._ctxs and self._devices instead of overriding this variables.
  • Check if using GPU or CPU before calling DataParallel, since it always uses GPU.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

zolastro avatar Dec 20 '22 14:12 zolastro