ViperEkura b092316385 feat : add distributed checkpoint via executor checkpoint_context ...

- Add checkpoint_context context manager to BaseExecutor with entry/exit barrier
- Add _gather_state_dict hook overridden per executor (template method)
- DDPExecutor skips unwrap on non-rank-0 to avoid redundant state_dict gather
- FSDPExecutor uses rank0_only=True to reduce memory on non-writers
- Remove redundant rank-0 guard from Checkpoint.save and manual barrier from Callback

2026-07-13 12:27:09 +08:00

..

__init__.py

feat: 新增FSDP并行后端

2026-05-25 19:43:14 +08:00

executor.py

feat : add distributed checkpoint via executor checkpoint_context

2026-07-13 12:27:09 +08:00

module.py

refactor: 优化参数传递，清理导入样式

2026-04-03 22:06:32 +08:00

setup.py

fix: auto-assign free port in spawn_parallel_fn to avoid EADDRINUSE

2026-07-12 19:21:56 +08:00