mirror of
https://github.com/wassname/pytorch-transformer-ts.git
synced 2026-06-27 18:06:14 +08:00
Kashif Rasul
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",
|
|
]
|