Skip to content
Snippets Groups Projects
Commit 6f882569 authored by Xuan Gu's avatar Xuan Gu
Browse files

Update README.md

parent 2623d8ab
No related branches found
No related tags found
No related merge requests found
......@@ -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.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment