Make the device an argument of the runner. #8
Description
The device on which the training of the runner is performed should by settable as an argument in the run() method. For both frameworks. This allows for more flexibe hardware usage.
{{ message }}