class_docstr: Allows dynamic batch training, extends pooling batcher with a scheduler
  config, which specifies how batch size should increase
class_docstr_tok: ["Allows","dynamic","batch","training",",","extends","pooling","batcher","with","a","scheduler","config",",","which","specifies","how","batch","size","should","increase"]
code: def get_batch_size(self, stage: Stage) -> int:
      if stage == Stage.TRAIN:
          print(f"using dynamic batch size {self.curr_batch_size}")
          return self.curr_batch_size
      else:
          return self._batch_sizes[stage]
  
imports: import math
  from typing import Iterable
  from pytext.common.constants import Stage
  from pytext.config.module_config import ModuleConfig
  from .data import PoolingBatcher
  from .sources import RawExample
label_desc: Use image node_id 6 for calling the DynamicPoolingBatcher obj's underlying member method code with example usage: obj.get_batch_size(stage) and returns: self, self