mirror of
https://github.com/hanzoai/gym-v1.git
synced 2026-10-30 18:35:21 +00:00
24 lines
597 B
Python
24 lines
597 B
Python
# simulator/__init__.py
|
|
|
|
from .strategy import Strategy
|
|
from .diloco import DiLoCoStrategy, DiLoCoCommunicator
|
|
from .optim import OptimSpec
|
|
from .sparta import SPARTAStrategy
|
|
from .federated_averaging import FedAvgStrategy
|
|
from .communicate_optimize_strategy import CommunicateOptimizeStrategy
|
|
|
|
from .sparta_diloco import SPARTADiLoCoStrategy
|
|
from .demo import DeMoStrategy
|
|
|
|
__all__ = [
|
|
"Strategy",
|
|
"DiLoCoStrategy",
|
|
"DiLoCoCommunicator",
|
|
"OptimSpec",
|
|
"SPARTAStrategy",
|
|
"FedAvgStrategy",
|
|
"CommunicateOptimizeStrategy",
|
|
"SPARTADiLoCoStrategy",
|
|
"DeMoStrategy",
|
|
]
|