fixed a bug

eth-easl · Feb 17, 2025 · 72184f3 · 72184f3
1 parent dca05a3
commit 72184f3
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 10 deletions.
diff --git a/modyn/trainer_server/internal/trainer/pytorch_trainer.py b/modyn/trainer_server/internal/trainer/pytorch_trainer.py
@@ -336,15 +336,6 @@ def train(self) -> None:  # pylint: disable=too-many-locals, too-many-branches
                         stopw.stop("Checkpoint")
                     if self._record_loss_every > 0 and trained_batches % self._record_loss_every == 0:
                         training_loss.append(loss.item())
-                        print(loss.item())
-                        # Log loss and batch number
-                        log_file = self._checkpoint_path / "training_log.txt"
-                        with (
-                            open(log_file, "a") as f  # pylint: disable=unspecified-encoding
-                        ):  # 'a' mode appends if the file exists, else creates it
-                            f.write(f"{trained_batches},{loss.item()}\n")
-                        # Example: Logging training losses in a loop
-
                     self._num_samples += len(sample_ids)
 
                     stopw.start("OnBatchEnd", resume=True)

diff --git a/modyn/trainer_server/internal/utils/training_info.py b/modyn/trainer_server/internal/utils/training_info.py
@@ -57,7 +57,6 @@ def __init__(
 
         self.shuffle = request.shuffle
         self.enable_accurate_gpu_measurements = request.enable_accurate_gpu_measurements
-        self.generative = request.generative
         assert (
             self.pretrained_model_path or not self.use_pretrained_model
         ), "Inconsistent pretrained model configuration"