You are viewing a plain text version of this content. The canonical link for it is here.
Posted to github@beam.apache.org by GitBox <gi...@apache.org> on 2022/08/12 14:06:50 UTC
[GitHub] [beam] yeandy opened a new issue, #22711: [Feature Request]: Add warning to logs for automatic GPU to CPU conversion
yeandy opened a new issue, #22711:
URL: https://github.com/apache/beam/issues/22711
### What would you like to happen?
In PytorchModelHandler, we do automatic conversion of the model from GPU to CPU if either device is not set to `GPU` or an GPU device is not found.
```
if device == 'GPU' and torch.cuda.is_available():
self._device = torch.device('cuda')
else:
self._device = torch.device('cpu')
```
We should print a warning to the logs when we do the conversion to inform users of the conversion.
### Issue Priority
Priority: 2
### Issue Component
Component: sdk-py-core
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: github-unsubscribe@beam.apache.org.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org
[GitHub] [beam] tvalentyn closed issue #22711: [Feature Request]: Add warning to logs for automatic GPU to CPU conversion
Posted by GitBox <gi...@apache.org>.
tvalentyn closed issue #22711: [Feature Request]: Add warning to logs for automatic GPU to CPU conversion
URL: https://github.com/apache/beam/issues/22711
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: github-unsubscribe@beam.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org