Fix Codacy Warnings (#477)
--------- Co-authored-by: Dario Coscia <dariocos99@gmail.com>
This commit is contained in:
committed by
Nicola Demo
parent
e3790e049a
commit
4177bfbb50
@@ -7,8 +7,20 @@ from .optimizer_interface import Optimizer
|
||||
|
||||
|
||||
class TorchOptimizer(Optimizer):
|
||||
"""
|
||||
TODO
|
||||
|
||||
:param Optimizer: _description_
|
||||
:type Optimizer: _type_
|
||||
"""
|
||||
|
||||
def __init__(self, optimizer_class, **kwargs):
|
||||
"""
|
||||
TODO
|
||||
|
||||
:param optimizer_class: _description_
|
||||
:type optimizer_class: _type_
|
||||
"""
|
||||
check_consistency(optimizer_class, torch.optim.Optimizer, subclass=True)
|
||||
|
||||
self.optimizer_class = optimizer_class
|
||||
@@ -16,6 +28,12 @@ class TorchOptimizer(Optimizer):
|
||||
self._optimizer_instance = None
|
||||
|
||||
def hook(self, parameters):
|
||||
"""
|
||||
TODO
|
||||
|
||||
:param parameters: _description_
|
||||
:type parameters: _type_
|
||||
"""
|
||||
self._optimizer_instance = self.optimizer_class(
|
||||
parameters, **self.kwargs
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user