From 0d8e6201f61319e84e226dcd145bd70c50406512 Mon Sep 17 00:00:00 2001 From: Wenqi Li Date: Thu, 26 May 2022 23:42:03 +0100 Subject: [PATCH] remove deprecated mentioning Signed-off-by: Wenqi Li --- acceleration/distributed_training/unet_evaluation_workflows.py | 2 +- modules/dynunet_pipeline/evaluator.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/acceleration/distributed_training/unet_evaluation_workflows.py b/acceleration/distributed_training/unet_evaluation_workflows.py index be817fd245..847fb12e5b 100644 --- a/acceleration/distributed_training/unet_evaluation_workflows.py +++ b/acceleration/distributed_training/unet_evaluation_workflows.py @@ -31,7 +31,7 @@ - Wrap the model with `DistributedDataParallel` after moving to expected device. - Put model file on every node, then load and map to expected GPU device in every process. - Wrap Dataset with `DistributedSampler`, disable the `shuffle` in sampler and DataLoader. -- Add `StatsHandler` and `SegmentationSaver` to the master process which is `dist.get_rank() == 0`. +- Add `StatsHandler` to the master process which is `dist.get_rank() == 0`. - ignite can automatically reduce metrics for distributed evaluation, refer to: https://github.com/pytorch/ignite/blob/v0.4.2/ignite/metrics/metric.py#L507 diff --git a/modules/dynunet_pipeline/evaluator.py b/modules/dynunet_pipeline/evaluator.py index aa198a5f64..184e422f63 100644 --- a/modules/dynunet_pipeline/evaluator.py +++ b/modules/dynunet_pipeline/evaluator.py @@ -42,7 +42,7 @@ class DynUNetEvaluator(SupervisedEvaluator): checkpoint into files. additional_metrics: more Ignite metrics that also attach to Ignite Engine. val_handlers: every handler is a set of Ignite Event-Handlers, must have `attach` function, like: - CheckpointHandler, StatsHandler, SegmentationSaver, etc. + CheckpointHandler, StatsHandler, etc. amp: whether to enable auto-mixed-precision evaluation, default is False. tta_val: whether to do the 8 flips (8 = 2 ** 3, where 3 represents the three dimensions) test time augmentation, default is False.