From 41b51959fc3a0aa4ce3128050d87588a07d6a4ea Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?M=C3=A9lodie=20Boillet?= <boillet@teklia.com>
Date: Thu, 8 Jun 2023 10:56:01 +0200
Subject: [PATCH] Merge zero optimizers functions
---
dan/manager/training.py | 5 +----
1 file changed, 1 insertion(+), 4 deletions(-)
diff --git a/dan/manager/training.py b/dan/manager/training.py
index 72acce30..f5779a14 100644
--- a/dan/manager/training.py
+++ b/dan/manager/training.py
@@ -532,10 +532,7 @@ class GenericTrainingManager:
def zero_optimizers(self, set_to_none=True):
for model_name in self.optimizers:
- self.zero_optimizer(model_name, set_to_none)
-
- def zero_optimizer(self, model_name, set_to_none=True):
- self.optimizers[model_name].zero_grad(set_to_none=set_to_none)
+ self.optimizers[model_name].zero_grad(set_to_none=set_to_none)
def train(self, mlflow_logging=False):
"""
--
GitLab