code: def state_dict(self):
      return {
          "current_momemtum": self.bn_lambda(self.last_epoch),
          "last_epoch": self.last_epoch,
      }
  
file_path: torch-points3d/torch_points3d/core/schedulers/bn_schedulers.py
imports: from typing import *
  from omegaconf import OmegaConf
  from torch import nn
  import logging
label_desc: Use image node_id 6 for calling the BNMomentumScheduler obj's underlying member method code with example usage: obj.state_dict() and returns: dict