diff --git a/README.md b/README.md index e86e280a63ca272b5d45e5ff18a84cdddb98fbd3..7b73e798a47c34cb0a1c4e18b3f5d5e76364acf6 100644 --- a/README.md +++ b/README.md @@ -113,3 +113,4 @@ Ref: https://pytorch-lightning.readthedocs.io/en/stable/common/trainer.html - For multi-node benchmarking, we need to use "srun" command; also, the line "#SBATCH --ntasks-per-node=8" has to be added. Otherwise the process will hang. - Use as large batch_size as possible for a more stable benchmark result. For single node, use 256; for multi-node, use 128. - Benchmarking with dim = 2, nodes = 1, gpus = 8, batch_size = 128, 256 takes ~2mins. +- Set path for enroot, see this [page](https://gitlab.liu.se/xuagu37/run-pytorch-and-tensorflow-containers-with-nvidia-enroot#set-path-to-user-container-storage).