-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathtest.sh
13 lines (12 loc) · 1011 Bytes
/
test.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
# Semanticist testing
accelerate launch --config_file=configs/onenode_config.yaml test_net.py --model ./output/tokenizer/models_xl --step 250000 --cfg_value 3.0 --test_num_slots 256
# or use torchrun
torchrun --nproc-per-node=8 test_net.py --model ./output/tokenizer/models_xl --step 250000 --cfg_value 3.0 --test_num_slots 256
# or use submitit
python submitit_test.py --ngpus=8 --nodes=1 --partition=xxx --model ./output/tokenizer/models_xl --step 250000 --cfg_value 3.0 --test_num_slots 256
# ϵLlamaGen testing
accelerate launch --config_file=configs/onenode_config.yaml test_net.py --model ./output/autoregressive/models_xl --step 250000 --cfg_value 5.0 --ae_cfg 1.0 --test_num_slots 32
# or use torchrun
torchrun --nproc-per-node=8 test_net.py --model ./output/autoregressive/models_xl --step 250000 --step 250000 --cfg_value 5.0 --ae_cfg 1.0 --test_num_slots 32
# or use submitit
python submitit_test.py --ngpus=8 --nodes=1 --partition=xxx --step 250000 --cfg_value 5.0 --ae_cfg 1.0 --test_num_slots 32