Skip to content

Made torch device configurable

Tim Tobias Bauerle requested to merge configurable_torch_device into wip

Made torch device configurable for each device. Provided default implementation so defining the torch device is not mandatory. Defaults to cuda:0 if cuda is available and CPU otherwise.

Merge request reports