File size: 610 Bytes
a7050c3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
#!/bin/bash

    # for DAMPENING_FRAC 0.01 0.1;
    # do
    #     for OBSERVER minmax mse;
    #     do
    #         for ACTORDER False group;
    #         do

# export DAMPENING_FRAC=0.01
# export OBSERVER="minmax"

export CUDA_VISIBLE_DEVICES=${1}
export MDL=${2}
export OBSERVER=${3} # minmax mse
export DAMPENING_FRAC=${4} # 0.01 0.1

for CALIB_SIZE in 128 512 1024;
do
    python w8a8.py --model_path ${MDL} --quant_path "output_dir_w8a8/${MDL}/calib${CALIB_SIZE}_eosFalse_damp${DAMPENING_FRAC}_obs${OBSERVER}" --calib_size ${CALIB_SIZE} --dampening_frac ${DAMPENING_FRAC} --observer ${OBSERVER}
done