code: def to(
      self, device: torch.device, available_devices: int = 1
  ) -> torch.nn.Module:
      self.net = self.net.to(device)
      if available_devices > 1:
          self.dp_wrapper_net = torch.nn.DataParallel(self.net)
      else:
          self.dp_wrapper_net = self.net
  
      self.device = device
      self.available_devices = available_devices
      return self
  
file_path: lightwood/lightwood/mixer/helpers/residual_net.py
imports: from typing import List
  import torch
  from torch import nn
  from lightwood.helpers.torch import LightwoodAutocast
  from lightwood.helpers.device import get_device_from_name, get_devices
  from lightwood.helpers.log import log
label_desc: Use image node_id 2 for calling the ResidualNet obj's underlying member method code with example usage: obj.to(device, available_devices) and returns: self