julius/prototorch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update component initializers

Browse Source
This commit is contained in:
Jensun Ravichandran
2021-05-12 16:28:55 +02:00
parent acd4ac6a86
commit 36a330aa66
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
prototorch/components/initializers.py
View File

@@ -101,6 +101,10 @@ class ClassAwareInitializer(ComponentsInitializer):
] ]
return torch.vstack(samples_list) return torch.vstack(samples_list)
def __del__(self):
del self.data
del self.labels
class StratifiedMeanInitializer(ClassAwareInitializer): class StratifiedMeanInitializer(ClassAwareInitializer):
def __init__(self, arg): def __init__(self, arg):