Reformulate syn-math

2021-05-09 18:53:18 +08:00
parent 853a702926
commit 58dee23a11
8 changed files with 57 additions and 32 deletions
--- a/lib/xlayers/super_module.py
+++ b/lib/xlayers/super_module.py
@@ -57,7 +57,7 @@ class TensorContainer:

    def requires_grad_(self, requires_grad=True):
        for tensor in self._tensors:
-          tensor.requires_grad_(requires_grad)
+            tensor.requires_grad_(requires_grad)

    @property
    def tensors(self):