Prototype MAML

2021-05-10 01:02:38 +08:00
parent 6e7b1c551f
commit cbd2afb4ef
14 changed files with 1497 additions and 702 deletions
--- a/lib/xlayers/super_module.py
+++ b/lib/xlayers/super_module.py
@@ -59,6 +59,9 @@ class TensorContainer:
        for tensor in self._tensors:
            tensor.requires_grad_(requires_grad)

+    def parameters(self):
+        return self._tensors
+
    @property
    def tensors(self):
        return self._tensors