wehub-resource-sync
34 lines
767 B
Python
34 lines
767 B
Python
# isort: off
|
|
try:
|
|
import lightning # noqa: F401
|
|
except ModuleNotFoundError:
|
|
try:
|
|
import pytorch_lightning # noqa: F401
|
|
except ModuleNotFoundError:
|
|
raise ModuleNotFoundError(
|
|
"PyTorch Lightning isn't installed. To install PyTorch Lightning, "
|
|
"please run 'pip install lightning'"
|
|
)
|
|
# isort: on
|
|
|
|
from ray.train.lightning._lightning_utils import (
|
|
RayDDPStrategy,
|
|
RayDeepSpeedStrategy,
|
|
RayFSDPStrategy,
|
|
RayLightningEnvironment,
|
|
RayTrainReportCallback,
|
|
prepare_trainer,
|
|
)
|
|
|
|
__all__ = [
|
|
"prepare_trainer",
|
|
"RayDDPStrategy",
|
|
"RayFSDPStrategy",
|
|
"RayDeepSpeedStrategy",
|
|
"RayLightningEnvironment",
|
|
"RayTrainReportCallback",
|
|
]
|
|
|
|
|
|
# DO NOT ADD ANYTHING AFTER THIS LINE.
|