以下是多折训练的代码:

model: type: NNUNet plan_path: data/preprocessed/nnUNetPlansv2.1_plans_3D.pkl stage: 0 cascade: True

train_dataset: type: MSDMultiFoldDataset plans_name: nnUNetPlansv2.1_plans_3D.pkl dataset_root: / result_dir: / raw_data_dir: data/raw_data decathlon_dir: data/decathlon cropped_data_dir: data/cropped preprocessed_dir: data/preprocessed plan2d: False plan3d: True num_batches_per_epoch: 250 num_folds: 5 current_fold: 2 stage: 0 unpack_data: True cascade: True mode: train

val_dataset: type: MSDMultiFoldDataset plans_name: nnUNetPlansv2.1_plans_3D.pkl dataset_root: / result_dir: / raw_data_dir: data/raw_data decathlon_dir: data/decathlon cropped_data_dir: data/cropped preprocessed_dir: data/preprocessed num_batches_per_epoch: 50 num_folds: 5 current_fold: 2 stage: 0 plan2d: False plan3d: True unpack_data: True cascade: True mode: val

optimizer: type: sgd momentum: 0.99 weight_decay: 0.00003 use_nesterov: True lr_scheduler: type: PolynomialDecay learning_rate: 0.01 end_lr: 0 power: 0.9 loss: types: - type: MultipleLoss plan_path: data/preprocessed/nnUNetPlansv2.1_plans_3D.pkl stage: 0 losses: - type: DC_and_CE_loss do_bg: False batch_dice: False coef: [1.0] coef: [1

model type NNUNet plan_path datapreprocessednnUNetPlansv21_plans_3Dpkl stage 0 cascade Truetrain_dataset type MSDDataset plans_name nnUNetPlansv21_plans_3Dpkl dataset_root result_dir raw_da

原文地址: https://www.cveoy.top/t/topic/cFJP 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录