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: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
