Merge pull request #37 from MayDomine/main

add __iter__ to make TransformerBlockList Iterable

Author: a710128

bmtrain/block_layer.py

@@ -831,7 +831,8 @@ def __init__(self, modules: Iterable[CheckpointBlock], sqrt=False) -> None:
 
     def __len__(self) -> int:
         return len(self._modules)
-    
+    def __iter__(self) -> Iterator[CheckpointBlock]:
+        return iter(self._modules.values())
     def __getitem__(self, index: Union[int, str]) -> CheckpointBlock:
         return self._modules[str(index)]