sync

.gitignore

@@ -121,11 +121,12 @@ sanbox.ipynb
 # s3 files
 config-docker.yaml
 config-env.yaml
-train_s3.sh
+config-dnd.yaml
+datasphere_run.sh
 build-and-push.yc.sh
+dnd_engine.py
 Dockerfile
 .dockerignore
 yc_logs_[0-3][0-9].[0-1][0-9].[0-9][0-9][0-9][0-9]-[0-2][0-9]:[0-5][0-9]:[0-5][0-9]/
-s3_train.py
 create_dist.sh
 vm-yc/

nnunetv2/training/nnUNetTrainer/variants/training_length/nnUNetTrainer_Xepochs.py

@@ -71,25 +71,17 @@ def __init__(self, plans: dict, configuration: str, fold: int, dataset_json: dic
         super().__init__(plans, configuration, fold, dataset_json, unpack_dataset, device)
         self.num_epochs = 500
 
-class nnUNetTrainer_1000epochs(nnUNetTrainer):
-    def __init__(self, plans: dict, configuration: str, fold: int, dataset_json: dict, unpack_dataset: bool = True,
-                 device: torch.device = torch.device('cuda')):
-        super().__init__(plans, configuration, fold, dataset_json, unpack_dataset, device)
-        self.num_epochs = 1000
-
-class nnUNetTrainer_500epochs(nnUNetTrainer):
+class nnUNetTrainer_750epochs(nnUNetTrainer):
     def __init__(self, plans: dict, configuration: str, fold: int, dataset_json: dict, unpack_dataset: bool = True,
                  device: torch.device = torch.device('cuda')):
         super().__init__(plans, configuration, fold, dataset_json, unpack_dataset, device)
-        self.num_epochs = 500
-
+        self.num_epochs = 750
 
-class nnUNetTrainer_750epochs(nnUNetTrainer):
+class nnUNetTrainer_1000epochs(nnUNetTrainer):
     def __init__(self, plans: dict, configuration: str, fold: int, dataset_json: dict, unpack_dataset: bool = True,
                  device: torch.device = torch.device('cuda')):
         super().__init__(plans, configuration, fold, dataset_json, unpack_dataset, device)
-        self.num_epochs = 750
-
+        self.num_epochs = 1000
 
 class nnUNetTrainer_2000epochs(nnUNetTrainer):
     def __init__(self, plans: dict, configuration: str, fold: int, dataset_json: dict, unpack_dataset: bool = True,