cleaning up demos (#312)

* cleaning up demos

* Update job_submit.sh

* Update README.md
This commit is contained in:
William Falcon
2019-10-05 14:48:22 -04:00
committed by GitHub
parent cdfcb01073
commit f7d762416c
2 changed files with 4 additions and 4 deletions
+2 -2
View File
@@ -6,5 +6,5 @@ To run this demo which launches a single job that trains on 2 nodes (2 gpus per
2. Create a conda environment with Lightning and a GPU PyTorch version.
3. Submit this script.
```bash
sbatch job_submit.sh your_env_name_with_lightning_installed
```
sbatch job_submit.sh --env=YourEnv
```
+2 -2
View File
@@ -8,7 +8,7 @@
#SBATCH --time=0-02:00:00
# activate conda env
conda activate $1
source activate $env
# -------------------------
# debugging flags (optional)
@@ -24,4 +24,4 @@ conda activate $1
# -------------------------
# run script from above
python multi_node_demo.py
srun python multi_node_demo.py