From f7d762416cc2c67f7f71f93f7adb14a7c12b8edd Mon Sep 17 00:00:00 2001
From: William Falcon <waf2107@columbia.edu>
Date: Sat, 5 Oct 2019 14:48:22 -0400
Subject: [PATCH] cleaning up demos (#312)

* cleaning up demos

* Update job_submit.sh

* Update README.md
---
 examples/multi_node_examples/README.md     | 4 ++--
 examples/multi_node_examples/job_submit.sh | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/examples/multi_node_examples/README.md b/examples/multi_node_examples/README.md
index d1d194f0..0db9d462 100644
--- a/examples/multi_node_examples/README.md
+++ b/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
-```
\ No newline at end of file
+sbatch job_submit.sh --env=YourEnv
+```
diff --git a/examples/multi_node_examples/job_submit.sh b/examples/multi_node_examples/job_submit.sh
index ee737198..176bf09f 100755
--- a/examples/multi_node_examples/job_submit.sh
+++ b/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
\ No newline at end of file
+srun python multi_node_demo.py