From 36a330aa66f7f0e7ca409cd4edcd4e1286a84634 Mon Sep 17 00:00:00 2001
From: Jensun Ravichandran <jjensun@gmail.com>
Date: Wed, 12 May 2021 16:28:55 +0200
Subject: [PATCH] Update component initializers

---
 prototorch/components/initializers.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/prototorch/components/initializers.py b/prototorch/components/initializers.py
index d94f9f1..4d55036 100644
--- 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):