Update component initializers

2021-05-12 16:28:55 +02:00
parent acd4ac6a86
commit 36a330aa66
1 changed files with 4 additions and 0 deletions
--- a/prototorch/components/initializers.py
+++ b/prototorch/components/initializers.py
@@ -101,6 +101,10 @@ class ClassAwareInitializer(ComponentsInitializer):
        ]
        return torch.vstack(samples_list)

+    def __del__(self):
+        del self.data
+        del self.labels
+

 class StratifiedMeanInitializer(ClassAwareInitializer):
    def __init__(self, arg):