add param group

2020-12-14 23:15:46 +08:00 · 2020-12-14 23:15:46 +08:00 · 88b08ab6cb
parent 358e4b1b82
commit 88b08ab6cb
1 changed files with 3 additions and 0 deletions
--- a/python/jittor/optim.py
+++ b/python/jittor/optim.py
@ -33,6 +33,9 @@ class Optimizer(object):
            assert isinstance(pg, dict)
            self.param_groups.append(pg)
        self.n_step = 0
+
+    def add_param_group(self, group):
+        self.param_groups.append(group)
    
    @property
    def defaults(self):