mirror of
https://github.com/wassname/pytorch-transformer-ts.git
synced 2026-07-15 11:25:26 +08:00
10 lines
275 B
Python
10 lines
275 B
Python
from .estimator import SwitchTransformerEstimator
|
|
from .lightning_module import SwitchTransformerLightningModule
|
|
from .module import SwitchTransformerModel
|
|
|
|
__all__ = [
|
|
"SwitchTransformerModel",
|
|
"SwitchTransformerLightningModule",
|
|
"SwitchTransformerEstimator",
|
|
]
|