From 0b2c6180de68a926aca4ad1df692944a7d504d70 Mon Sep 17 00:00:00 2001 From: Chris Rinard Date: Wed, 23 Aug 2023 11:32:58 -0700 Subject: [PATCH] reee --- scripts/train/benchmarking/run_local.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/train/benchmarking/run_local.sh b/scripts/train/benchmarking/run_local.sh index 3ca57f0840..1a881f10b6 100755 --- 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 \ No newline at end of file +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 \ No newline at end of file