accelerator: Create an accelerator

In mlverse/torchlight: Higher Level 'API' for 'torch'

Create an accelerator

Description

Create an accelerator

Usage

accelerator(
  device_placement = TRUE,
  cpu = FALSE,
  cuda_index = torch::cuda_current_device()
)

Arguments

device_placement	(logical) whether the accelerator object should handle device placement. Default: TRUE
cpu	(logical) whether the training procedure should run on the CPU.
cuda_index	(integer) index of the CUDA device to use if multiple GPUs are available. Default: the result of torch::cuda_current_device().

mlverse/torchlight documentation built on Sept. 19, 2024, 11:22 p.m.