From a889b37beb2e289093010cf7731075ee79fb1b48 Mon Sep 17 00:00:00 2001 From: Roman Fitzjalen Date: Sat, 26 Oct 2024 11:33:49 +0300 Subject: [PATCH] add autopet finetune trainer --- nnunetv2/training/nnUNetTrainer/autoPET3_Trainer.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/nnunetv2/training/nnUNetTrainer/autoPET3_Trainer.py b/nnunetv2/training/nnUNetTrainer/autoPET3_Trainer.py index bf49f1e4..538092bc 100644 --- a/nnunetv2/training/nnUNetTrainer/autoPET3_Trainer.py +++ b/nnunetv2/training/nnUNetTrainer/autoPET3_Trainer.py @@ -535,4 +535,10 @@ def perform_actual_validation(self, save_probabilities: bool = False): also_print_to_console=True) self.set_deep_supervision_enabled(True) - compute_gaussian.cache_clear() \ No newline at end of file + compute_gaussian.cache_clear() + + +class nnUNetTrainerFineTuneLR1e3_619autoPetStem_010misalign_rot555_sh022_noSmooth_organs_1500ep(autoPET3_Trainer): + def __init__(self, plans, configuration, fold, dataset_json, unpack_dataset = True, device = torch.device('cuda')): + super().__init__(plans, configuration, fold, dataset_json, unpack_dataset, device) + self.num_epochs = 1500