reee

mosaicml · Aug 23, 2023 · 0b2c618 · 0b2c618
1 parent 80a9bca
commit 0b2c618
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/train/benchmarking/run_local.sh b/scripts/train/benchmarking/run_local.sh
@@ -7,4 +7,4 @@ IMAGE="mosaicml/pytorch:2.0.1_cu118-python3.10-ubuntu20.04"
 # A100 40GB
 CLUSTER_40GB=r1z1
 # seqlen 2048
-python submit_benchmarks.py --project $PROJECT -m 125m.yaml -g 8 --microbatch_size  24 --accum  3 --image $IMAGE --git_commit $GIT_COMMIT --gpu_type a100_40gb --cluster $CLUSTER_40GB -s 11 11 --RUN --LOCAL
+python submit_benchmarks.py --project $PROJECT -m 125m.yaml -g 8 --microbatch_size  24 --accum  3 --image $IMAGE --git_commit $GIT_COMMIT --gpu_type a100_80gb --cluster $CLUSTER_40GB -s 11 11 --RUN --LOCAL