From 9e68173d7c6aa2bde0a338de9321ec8ce10348b7 Mon Sep 17 00:00:00 2001 From: Fabian Isensee Date: Thu, 18 Apr 2024 14:38:01 +0200 Subject: [PATCH] remove train/val time prints --- nnunetv2/training/nnUNetTrainer/nnUNetTrainer.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/nnunetv2/training/nnUNetTrainer/nnUNetTrainer.py b/nnunetv2/training/nnUNetTrainer/nnUNetTrainer.py index 2e873332a..decda30c0 100644 --- a/nnunetv2/training/nnUNetTrainer/nnUNetTrainer.py +++ b/nnunetv2/training/nnUNetTrainer/nnUNetTrainer.py @@ -1291,7 +1291,6 @@ 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) @@ -1299,7 +1298,6 @@ def run_training(self): 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)