Update DEBUG INFO

2021-05-10 14:14:06 +08:00
parent 147da98f94
commit 0dbbc286c9
8 changed files with 536 additions and 274 deletions
--- a/lib/xlayers/super_module.py
+++ b/lib/xlayers/super_module.py
@@ -46,6 +46,13 @@ class TensorContainer:
            result.append(name, new_tensor, self._param_or_buffers[index])
        return result

+    def create_container(self, tensors):
+        result = TensorContainer()
+        for index, name in enumerate(self._names):
+            new_tensor = tensors[index]
+            result.append(name, new_tensor, self._param_or_buffers[index])
+        return result
+
    def no_grad_clone(self):
        result = TensorContainer()
        with torch.no_grad():