class_docstr: Base class for any torch model.
  
  :param device: Device the networks should be located on (cpu or cuda)
class_docstr_tok: ["Base","class","for","any","torch","model",".",":","param","device",":","Device","the","networks","should","be","located","on","(","cpu","or","cuda",")"]
code: def __init__(self, device: str):
      self._device = device
      self.to(self._device)
  
file_path: maze/maze/core/agent/torch_model.py
imports: from abc import abstractmethod
  from typing import List, Dict
  import torch
label_desc: Use image node_id 1 to create a new TorchModel object  with example: obj = TorchModel(device)