This commit is contained in:
Nicola Demo
2024-08-01 16:30:35 +02:00
parent b7d512e8bf
commit 686b557144

View File

@@ -18,4 +18,4 @@ class TorchOptimizer(Optimizer):
def hook(self, parameters):
self.optimizer_instance = self.optimizer_class(
parameters, **self.kwargs
)
)