class_docstr: Stateful Tensor Manager, inspired from PatrickStar
  
  PatrickStar: Parallel Training of Pre-trained Models via Chunk-based Memory Management
  https://arxiv.org/abs/2108.05818
class_docstr_tok: ["Stateful","Tensor","Manager",",","inspired","from","PatrickStar","PatrickStar",":","Parallel","Training","of","Pre-trained","Models","via","Chunk-based","Memory","Management","https",":","\/\/arxiv.org\/abs\/2108.05818"]
docstr: This function must be called when each iteration finishes
docstr_tok: ["This","function","must","be","called","when","each","iteration","finishes"]
code: def finish_iter(self):
      """This function must be called when each iteration finishes"""
      self._warmup = False
      self._compute_idx = -1
      self._cpu_gpu_move_volume = 0
      self._layout_time = 0
      self._evict_time = 0
  
file_path: colossalai/colossalai/legacy/zero/gemini/stateful_tensor_mgr.py
imports: import functools
  import types
  from time import time
  from typing import List
  from colossalai.utils.device import get_current_device
  from .stateful_tensor import StatefulTensor, TensorState
  from .tensor_placement_policy import TensorPlacementPolicy
  from .tensor_utils import colo_model_data_tensor_move_inline
label_desc: Use image node_id 4 for calling the StatefulTensorMgr obj's underlying member method code with example usage: obj.finish_iter() without return types