reset loss_values df on refit

sdv-dev · Oct 11, 2023 · 199f932 · 199f932
1 parent 44e8881
commit 199f932
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/deepecho/models/par.py b/deepecho/models/par.py
@@ -327,6 +327,9 @@ def fit_sequences(self, sequences, context_types, data_types):
             pbar_description = 'Loss ({loss:.3f})'
             iterator.set_description(pbar_description.format(loss=0))
 
+        # Reset loss_values dataframe
+        self.loss_values = pd.DataFrame(columns=['Epoch', 'Loss'])
+
         X_padded, seq_len = torch.nn.utils.rnn.pad_packed_sequence(X)
         for epoch in iterator:
             Y = self._model(X, C)