class_docstr: Base container for PyTorch models.
class_docstr_tok: ["Base","container","for","PyTorch","models","."]
code: def to(self, device):
      self.model.to(device)
      return self
  
imports: import pickle
  import os
  import numpy
  import shutil
  import torch
  import warnings
  import hummingbird
  from hummingbird.ml._utils import pandas_installed, get_device, from_strings_to_ints, dump_versions, check_dumped_versions
  from hummingbird.ml.operator_converters import constants
  from hummingbird.ml.containers._sklearn_api_containers import SklearnContainer, SklearnContainerTransformer, SklearnContainerRegression, SklearnContainerClassification, SklearnContainerAnomalyDetection
label_desc: Use image node_id 3 for calling the PyTorchSklearnContainer obj's underlying member method code with example usage: obj.to(device) and returns: self