Skip to content

Commit 8efb8a4

Browse files
committed
Added experiment scripts for gemma-7b
1 parent 128830e commit 8efb8a4

7 files changed

+95
-2
lines changed

experiments/job-22841975-logs.out

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
Removing miniconda version 22.11.1-1
2+
Loading miniconda version 22.11.1-1
3+
python3: can't open file '/home/avsingh_umass_edu/COMPSCI-685-Project/experiments/src/reasoning/main.py': [Errno 2] No such file or directory
Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
#!/bin/bash
2+
#SBATCH -c 4 # Number of Cores per Task
3+
#SBATCH --mem=50G # Requested Memory
4+
#SBATCH -p gpu # Partition
5+
#SBATCH -G 1 # Number of GPUs
6+
#SBATCH --constraint vram48 # GPU Memory
7+
#SBATCH -t 12:00:00 # Job time limit
8+
#SBATCH -o job-%j-logs.out # %j = job ID
9+
10+
module load miniconda/22.11.1-1
11+
conda activate cs685
12+
13+
python3 src/reasoning/main.py --model "google/gemma-7b" --dataset "gsm8k" --dataset-size 8792 \
14+
--recurring-self-consistency \
15+
--num-shots 8 \
16+
--max-out-tokens 120 \
17+
--gpu-memory-utilization 0.9 \
18+
--output-file "results/recurring__gsm8k__gemma-7b__output-optimized.json"
Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
#!/bin/bash
2+
#SBATCH -c 4 # Number of Cores per Task
3+
#SBATCH --mem=50G # Requested Memory
4+
#SBATCH -p gpu # Partition
5+
#SBATCH -G 1 # Number of GPUs
6+
#SBATCH --constraint vram48 # GPU Memory
7+
#SBATCH -t 12:00:00 # Job time limit
8+
#SBATCH -o job-%j-logs.out # %j = job ID
9+
10+
module load miniconda/22.11.1-1
11+
conda activate cs685
12+
13+
python3 src/reasoning/main.py --model "google/gemma-7b" --dataset "gsm8k" --dataset-size 2198 --dataset-offset 0 \
14+
--recurring-self-consistency \
15+
--num-shots 8 \
16+
--max-out-tokens 120 \
17+
--gpu-memory-utilization 0.9 \
18+
--output-file "results/recurring__gsm8k__gemma-7b__output-optimized_1.json"
Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
#!/bin/bash
2+
#SBATCH -c 4 # Number of Cores per Task
3+
#SBATCH --mem=50G # Requested Memory
4+
#SBATCH -p gpu # Partition
5+
#SBATCH -G 1 # Number of GPUs
6+
#SBATCH --constraint vram48 # GPU Memory
7+
#SBATCH -t 12:00:00 # Job time limit
8+
#SBATCH -o job-%j-logs.out # %j = job ID
9+
10+
module load miniconda/22.11.1-1
11+
conda activate cs685
12+
13+
python3 src/reasoning/main.py --model "google/gemma-7b" --dataset "gsm8k" --dataset-size 2198 --dataset-offset 2198 \
14+
--recurring-self-consistency \
15+
--num-shots 8 \
16+
--max-out-tokens 120 \
17+
--gpu-memory-utilization 0.9 \
18+
--output-file "results/recurring__gsm8k__gemma-7b__output-optimized_2.json"
Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
#!/bin/bash
2+
#SBATCH -c 4 # Number of Cores per Task
3+
#SBATCH --mem=50G # Requested Memory
4+
#SBATCH -p gpu # Partition
5+
#SBATCH -G 1 # Number of GPUs
6+
#SBATCH --constraint vram48 # GPU Memory
7+
#SBATCH -t 12:00:00 # Job time limit
8+
#SBATCH -o job-%j-logs.out # %j = job ID
9+
10+
module load miniconda/22.11.1-1
11+
conda activate cs685
12+
13+
python3 src/reasoning/main.py --model "google/gemma-7b" --dataset "gsm8k" --dataset-size 2198 --dataset-offset 4396 \
14+
--recurring-self-consistency \
15+
--num-shots 8 \
16+
--max-out-tokens 120 \
17+
--gpu-memory-utilization 0.9 \
18+
--output-file "results/recurring__gsm8k__gemma-7b__output-optimized_3.json"
Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
#!/bin/bash
2+
#SBATCH -c 4 # Number of Cores per Task
3+
#SBATCH --mem=50G # Requested Memory
4+
#SBATCH -p gpu # Partition
5+
#SBATCH -G 1 # Number of GPUs
6+
#SBATCH --constraint vram48 # GPU Memory
7+
#SBATCH -t 2:00:00 # Job time limit
8+
#SBATCH -o job-%j-logs.out # %j = job ID
9+
10+
module load miniconda/22.11.1-1
11+
conda activate cs685
12+
13+
python3 src/reasoning/main.py --model "google/gemma-7b" --dataset "gsm8k" --dataset-size 2198 --dataset-offset 6594 \
14+
--recurring-self-consistency \
15+
--num-shots 8 \
16+
--max-out-tokens 120 \
17+
--gpu-memory-utilization 0.9 \
18+
--output-file "results/recurring__gsm8k__gemma-7b__output-optimized_4.json"

experiments/recurring_gemma-7b.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
#SBATCH -p gpu # Partition
55
#SBATCH -G 1 # Number of GPUs
66
#SBATCH --constraint vram80 # GPU Memory
7-
#SBATCH -t 23:59:59 # Job time limit
7+
#SBATCH -t 12:00:00 # Job time limit
88
#SBATCH -o job-%j-logs.out # %j = job ID
99

1010
module load miniconda/22.11.1-1
@@ -15,4 +15,4 @@ python3 src/reasoning/main.py --model "google/gemma-7b" --dataset "gsm8k" --data
1515
--num-shots 8 \
1616
--max-out-tokens 120 \
1717
--gpu-memory-utilization 0.9 \
18-
--output-file "results/recurring__gsm8k__gemma-7b__output.json"
18+
--output-file "results/recurring__gsm8k__gemma-7b__output.json"

0 commit comments

Comments
 (0)