auto_device {mlr3torch}R Documentation

Auto Device

Description

First tries cuda, then cpu.

Usage

auto_device(device = NULL)

Arguments

device

(character(1))
The device. If not NULL, is returned as is.

[Package mlr3torch version 0.1.0 Index]