code: def __init__(
      self,
      local_nn=[3, 64, 128, 1024],
      global_nn=[1024, 512, 256],
      batch_size=1,
  ):
      super().__init__(
          MiniPointNet(local_nn, global_nn),
          global_nn[-1],
          3,
          batch_size,
      )
  
imports: import torch
  from torch_geometric.nn import global_max_pool, global_mean_pool
  from torch_points3d.core.common_modules.base_modules import *
  from torch_points3d.core.common_modules.spatial_transform import BaseLinearTransformSTNkD
  from torch_points3d.models.base_model import BaseInternalLossModule
label_desc: Use image node_id 1 to create a new PointNetSTN3D object from inherited base classes: BaseLinearTransformSTNkD with example: obj = PointNetSTN3D(local_nn, global_nn, batch_size)