mirror of
https://github.com/wassname/pytorch-transformer-ts.git
synced 2026-06-27 18:06:14 +08:00
fixed names
This commit is contained in:
@@ -3,7 +3,7 @@ from .lightning_module import TransformerLightningModule
|
||||
from .module import TransformerModel
|
||||
|
||||
__all__ = [
|
||||
"TransformerModel",
|
||||
"TransformerLightningModule",
|
||||
"TransformerEstimator",
|
||||
"ReformerModel",
|
||||
"ReformerLightningModule",
|
||||
"ReformerEstimator",
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user