set_device: Set model device.

In krzjoa/torchts: Time series Models with torch

Set model device.

Description

Set model device.

Usage

set_device(object, device, ...)

Arguments

object	An neural network object.
device	(character) Selected device.

Value

Object of the same class with device set.

Examples

rnn_net <-
  model_rnn(
    input_size  = 1,
    output_size = 1,
    hidden_size = 10
  ) %>%
 set_device("cpu")

rnn_net

krzjoa/torchts documentation built on June 24, 2022, 5:30 a.m.