docstr: Reset the bucket storage after reduction, only release the tensors have been reduced
docstr_tok: ["Reset","the","bucket","storage","after","reduction",",","only","release","the","tensors","have","been","reduced"]
code: def reset(self):
      """Reset the bucket storage after reduction, only release the tensors have been reduced"""
      cur_offset = self.offset_list.pop(0)
      self._param_list = self._param_list[cur_offset:]
      self._padding_size = self._padding_size[cur_offset:]
      for _ in range(cur_offset):
          del self.grad_to_param_mapping[
              next(iter(self.grad_to_param_mapping))
          ]
      for rank in range(self._world_size):
          self._grad_in_bucket[rank] = self._grad_in_bucket[rank][
              cur_offset:
          ]
  
file_path: ColossalAI/colossalai/zero/low_level/bookkeeping/bucket_store.py
imports: from typing import Dict
  import torch
  from torch import Tensor
  from torch._utils import _flatten_dense_tensors
  from torch.distributed import ProcessGroup
  from .base_store import BaseStore
label_desc: Use image node_id 9 for calling the BucketStore obj's underlying member method code with example usage: obj.reset() without return types