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.


mlr-org/mlr3torch documentation built on April 17, 2025, 8:22 p.m.