Skip to content

Commit e8003d6

Browse files
authored
Update and rename 1 to __init__.py
1 parent f48e9a6 commit e8003d6

2 files changed

Lines changed: 41 additions & 1 deletion

File tree

src/core/1

Lines changed: 0 additions & 1 deletion
This file was deleted.

src/core/__init__.py

Lines changed: 41 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,41 @@
1+
"""
2+
Core distributed training components
3+
4+
This module contains the fundamental distributed training implementations:
5+
- DistributedTrainer: Base DDP/FSDP trainer
6+
- EnhancedDistributedTrainer: Production trainer with fault tolerance
7+
- CommunicationOptimizer: Communication optimization strategies
8+
"""
9+
10+
from .distributed_training import (
11+
DistributedTrainer,
12+
SimpleResNet,
13+
benchmark_training,
14+
)
15+
16+
from .enhanced_trainer import EnhancedDistributedTrainer
17+
18+
from .communication_optimizer import (
19+
CommunicationOptimizer,
20+
GradientAccumulator,
21+
OverlapCommunicator,
22+
benchmark_communication,
23+
)
24+
25+
__all__ = [
26+
# Trainers
27+
"DistributedTrainer",
28+
"EnhancedDistributedTrainer",
29+
30+
# Models
31+
"SimpleResNet",
32+
33+
# Optimization
34+
"CommunicationOptimizer",
35+
"GradientAccumulator",
36+
"OverlapCommunicator",
37+
38+
# Benchmarking
39+
"benchmark_training",
40+
"benchmark_communication",
41+
]

0 commit comments

Comments
 (0)