code: def set_input(self, data, device):
      self.batch_idx = data.batch.squeeze()
      coords = torch.cat(
          [data.batch.unsqueeze(-1).int(), data.coords.int()], -1
      )
      self.input = ME.SparseTensor(
          features=data.x, coordinates=coords, device=device
      )
      self.labels = data.y.to(device)
  
file_path: torch-points3d/torch_points3d/models/segmentation/minkowski.py
imports: import logging
  import torch.nn.functional
  import torch
  from torch_points3d.modules.MinkowskiEngine import *
  from torch_points3d.models.base_architectures import UnwrappedUnetBasedModel
  from torch_points3d.models.base_model import BaseModel
  from torch_points3d.datasets.segmentation import IGNORE_LABEL
  from torch_points3d.applications.minkowski import Minkowski
label_desc: Use image node_id 2 for calling the Minkowski_Baseline_Model obj's underlying member method code with example usage: obj.set_input(data, device) without return types