Learn R Programming

shapr (version 1.0.1)

vaeac_check_cuda: Function that checks for access to CUDA

Description

Function that checks for access to CUDA

Usage

vaeac_check_cuda(cuda)

Value

The function does not return anything.

Arguments

cuda

Logical (default is FALSE). If TRUE, then the vaeac model will be trained using cuda/GPU. If torch::cuda_is_available() is FALSE, the we fall back to use CPU. If FALSE, we use the CPU. Using a GPU for smaller tabular dataset often do not improve the efficiency. See vignette("installation", package = "torch") fo help to enable running on the GPU (only Linux and Windows).

Author

Lars Henry Berge Olsen