Skip to content

GitLab

Commit e815f118 authored by zhougaofeng's avatar zhougaofeng
Browse files

Update run.sh

parent 254d6a34
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 4 deletions
run.sh
View file @ e815f118
...@@ -30,7 +30,3 @@ CUDA_VISIBLE_DEVICES=0,1,2,3 python3 -m torch.distributed.launch --master_addr $ ...@@ -30,7 +30,3 @@ CUDA_VISIBLE_DEVICES=0,1,2,3 python3 -m torch.distributed.launch --master_addr $
--fsdp_transformer_layer_cls_to_wrap 'LlamaDecoderLayer' \ --fsdp_transformer_layer_cls_to_wrap 'LlamaDecoderLayer' \
--tf32 True --tf32 True
#精度测试（推荐使用八卡），若要改变卡数，需要到对应的py文件中修改代码
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
python eval_gsm8k.py --model $SAVE_PATH --data_path ./data/test/GSM8K_test.jsonl
python eval_math.py --model $SAVE_PATH --data_path ./data/test/MATH_test.jsonl
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment