class_docstr: A wrapper module that's more convenient to use.
code: def device(self):
      return next(self.parameters()).device
  
imports: import abc
  import logging
  import math
  import os
  import fire
  import torch
  import torchdiffeq
  import torchvision
  import tqdm
  from torch import nn, optim
  from torch.utils import data
  import torchsde
  from .None import unet
label_desc: Use image node_id 3 for calling the Module obj's underlying member method code with example usage: obj.device() and returns: next