diff --git a/README.md b/README.md index fe69aa40e983775d6cebb4e1bd7c7e98f760f9ed..52ca0b47b4f5ea9ee1df621c690f75d8834f2faa 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.