This commit is contained in:
Nicola Demo
2024-08-05 17:34:34 +02:00
parent 686b557144
commit 5245a0b68c
19 changed files with 483 additions and 173 deletions

11
pina/optim/__init__.py Normal file
View 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