fix bug

lilab-bcb · Jun 10, 2021 · ab041ba · ab041ba
1 parent 1c46d21
commit ab041ba
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nmf/_inmf_base.py b/nmf/_inmf_base.py
@@ -52,6 +52,8 @@ def _initialize_W_H_V(self):
         for k in range(self._n_batches):
             H = 2.0 * torch.rand((self.X[k].shape[0], self._n_components), dtype=self._tensor_dtype, device=self._device_type)
             V = 2.0 * torch.rand((self._n_components, self._n_features), dtype=self._tensor_dtype, device=self._device_type)
+            self.H.append(H)
+            self.V.append(V)
 
 
     def _trace(self, A, B):