cleaning up demos (#312)

* cleaning up demos

* Update job_submit.sh

* Update README.md

Author: williamFalcon

examples/multi_node_examples/README.md

@@ -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
+```

examples/multi_node_examples/job_submit.sh

@@ -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