refact
This commit is contained in:
11
pina/optim/__init__.py
Normal file
11
pina/optim/__init__.py
Normal file
@@ -0,0 +1,11 @@
|
||||
__all__ = [
|
||||
"Optimizer",
|
||||
"TorchOptimizer",
|
||||
"Scheduler",
|
||||
"TorchScheduler",
|
||||
]
|
||||
|
||||
from .optimizer_interface import Optimizer
|
||||
from .torch_optimizer import TorchOptimizer
|
||||
from .scheduler_interface import Scheduler
|
||||
from .torch_scheduler import TorchScheduler
|
||||
Reference in New Issue
Block a user