auto_device: Auto Device

View source: R/utils.R

auto_deviceR 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.


mlr3torch documentation built on April 4, 2025, 3:03 a.m.