remove train/val time prints

MIC-DKFZ · Apr 18, 2024 · 9e68173 · 9e68173
1 parent e890b9d
commit 9e68173
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/nnunetv2/training/nnUNetTrainer/nnUNetTrainer.py b/nnunetv2/training/nnUNetTrainer/nnUNetTrainer.py
@@ -1291,15 +1291,13 @@ def run_training(self):
 
             self.on_train_epoch_start()
             train_outputs = []
-            st = time()
             for batch_id in range(self.num_iterations_per_epoch):
                 train_outputs.append(self.train_step(next(self.dataloader_train)))
             self.on_train_epoch_end(train_outputs)
 
             with torch.no_grad():
                 self.on_validation_epoch_start()
                 val_outputs = []
-                st = time()
                 for batch_id in range(self.num_val_iterations_per_epoch):
                     val_outputs.append(self.validation_step(next(self.dataloader_val)))
                 self.on_validation_epoch_end(val_outputs)