reducing memory usage in deepseparation
@@ -108,6 +108,7 @@ class DeepSeparation(mask_separation_base.MaskSeparationBase):
"""
input_data = self._preprocess()
+ with torch.no_grad():
output = self.model(input_data)
if 'embedding' in output:
