Skip to content

Commit

Permalink
Skeleton Recall Trainer
Browse files Browse the repository at this point in the history
  • Loading branch information
mrokuss committed Jul 10, 2024
1 parent e8813c5 commit 8200fd2
Showing 1 changed file with 0 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -37,8 +37,6 @@
from batchgeneratorsv2.transforms.utils.seg_to_regions import ConvertSegmentationToRegionsTransform
from nnunetv2.training.data_augmentation.custom_transforms.skeletonization import SkeletonTransform

from nnunetv2.training.nnUNetTrainer.variants.data_augmentation.nnUNetTrainerNoMirroring import nnUNetTrainer_onlyMirror01

class nnUNetTrainerSkeletonRecall(nnUNetTrainer):
def __init__(self, plans: dict, configuration: str, fold: int, dataset_json: dict, unpack_dataset: bool = True,
device: torch.device = torch.device('cuda')):
Expand Down

0 comments on commit 8200fd2

Please sign in to comment.