folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Delete useless classes (move into optim folder)

Browse Source
This commit is contained in:
FilippoOlivo
2024-09-28 12:19:53 +02:00
committed by Nicola Demo
parent 2be57944ba
commit a529241580
2 changed files with 0 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
pina/optimizer.py
View File

@@ -1,21 +0,0 @@
""" Module for PINA Optimizer """
import torch
from .utils import check_consistency
class Optimizer: # TODO improve interface
pass
class TorchOptimizer(Optimizer):
def __init__(self, optimizer_class, **kwargs):
check_consistency(optimizer_class, torch.optim.Optimizer, subclass=True)
self.optimizer_class = optimizer_class
self.kwargs = kwargs
def hook(self, parameters):
self.optimizer_instance = self.optimizer_class(
parameters, **self.kwargs
)

29
pina/scheduler.py
View File

@@ -1,29 +0,0 @@
""" Module for PINA Scheduler """
try:
from torch.optim.lr_scheduler import LRScheduler # torch >= 2.0
except ImportError:
from torch.optim.lr_scheduler import (
_LRScheduler as LRScheduler,
) # torch < 2.0
from .optimizer import Optimizer
from .utils import check_consistency
class Scheduler: # TODO improve interface
pass
class TorchScheduler(Scheduler):
def __init__(self, scheduler_class, **kwargs):
check_consistency(scheduler_class, LRScheduler, subclass=True)
self.scheduler_class = scheduler_class
self.kwargs = kwargs
def hook(self, optimizer):
check_consistency(optimizer, Optimizer)
self.scheduler_instance = self.scheduler_class(
optimizer.optimizer_instance, **self.kwargs
)