From 6f882569b328baffa76eba0390f5329d33fbc748 Mon Sep 17 00:00:00 2001 From: Xuan Gu <xuan.gu@liu.se> Date: Wed, 9 Nov 2022 09:39:32 +0000 Subject: [PATCH] Update README.md --- README.md | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index fe69aa4..52ca0b4 100644 --- a/README.md +++ b/README.md @@ -95,11 +95,10 @@ Coefficient of variation is calculated as the ratio of the standard deviation to #### Notes -- Line 116 of /DeepLearningExamples/PyTorch/Segmentation/nnUNet/main.py should be changed: -```trainer.test(model, test_dataloaders=data_module.test_dataloader(), ckpt_path=ckpt_path)``` -to +- Line 116 of /DeepLearningExamples/PyTorch/Segmentation/nnUNet/main.py should be changed: +```trainer.test(model, test_dataloaders=data_module.test_dataloader(), ckpt_path=ckpt_path)``` +to ```trainer.test(model, dataloaders=data_module.test_dataloader(), ckpt_path=ckpt_path)``` - Ref: https://pytorch-lightning.readthedocs.io/en/stable/common/trainer.html - It seems running directly via singularity shell will give worse performance (when I WFH). We should run it via sbatch script instead. - It took around a week to finish 100 iterations of benchmarking for all sets of parameters. -- GitLab