close summary writer

Ard0rAI · Feb 20, 2020 · f70752b · f70752b
1 parent 8d70919
commit f70752b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/spert/spert_trainer.py b/spert/spert_trainer.py
@@ -125,6 +125,7 @@ def train(self, train_path: str, valid_path: str, types_path: str, input_reader_
 
         self._logger.info("Logged in: %s" % self._log_path)
         self._logger.info("Saved in: %s" % self._save_path)
+        self._summary_writer.close()
 
     def eval(self, dataset_path: str, types_path: str, input_reader_cls: BaseInputReader):
         args = self.args
@@ -162,7 +163,9 @@ def eval(self, dataset_path: str, types_path: str, input_reader_cls: BaseInputRe
 
         # evaluate
         self._eval(model, input_reader.get_dataset(dataset_label), input_reader)
+
         self._logger.info("Logged in: %s" % self._log_path)
+        self._summary_writer.close()
 
     def _train_epoch(self, model: torch.nn.Module, compute_loss: Loss, optimizer: Optimizer, dataset: Dataset,
                      updates_epoch: int, epoch: int):