DeepOBS
DeepOBS copied to clipboard
Make the device an argument of the runner
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 flexible hardware usage.