Skip to content
Snippets Groups Projects
Unverified Commit bb1f56d7 authored by i-robot's avatar i-robot Committed by Gitee
Browse files

!3836 移除lenet golden-stick pretrain

Merge pull request !3836 from KXiong/master
parents 96a7f8c8 fb1e6265
No related branches found
No related tags found
No related merge requests found
......@@ -307,10 +307,6 @@ bash run_standalone_train_gpu.sh ../quantization/simqat/ ../quantization/simqat/
# standalone training example, apply SimQAT and train from full precision checkpoint
cd ./golden_stick/scripts/
bash run_standalone_train_gpu.sh ../quantization/simqat/ ../quantization/simqat/lenet_mnist_config.yaml /path/to/dataset FP32 /path/to/fp32_ckpt
# standalone training example, apply SimQAT and train from pretrained checkpoint
cd ./golden_stick/scripts/
bash run_standalone_train_gpu.sh ../quantization/simqat/ ../quantization/simqat/lenet_mnist_config.yaml /path/to/dataset PRETRAINED /path/to/pretrained_ckpt
```
## Evaluation Process
......
......@@ -307,10 +307,6 @@ bash run_standalone_train_gpu.sh ../quantization/simqat/ ../quantization/simqat/
# 训练示例(应用SimQAT算法并加载预训练的全精度checkoutpoint,并进行量化训练)
cd ./golden_stick/scripts/
bash run_standalone_train_gpu.sh ../quantization/simqat/ ../quantization/simqat/lenet_mnist_config.yaml /path/to/dataset FP32 /path/to/fp32_ckpt
# 训练示例(应用SimQAT算法并加载上次量化训练的checkoutpoint,继续进行量化训练)
cd ./golden_stick/scripts/
bash run_standalone_train_gpu.sh ../quantization/simqat/ ../quantization/simqat/lenet_mnist_config.yaml /path/to/dataset PRETRAINED /path/to/pretrained_ckpt
```
## 评估过程
......
......@@ -89,9 +89,6 @@ cd ./train || exit
if [ "x$CKPT_TYPE" == "xFP32" ]; then
python train.py --config_path=$CONFIG_FILE --device_target="GPU" --data_path=$DATASET_PATH \
--fp32_ckpt=$CKPT_FILE &> log &
elif [ "x$CKPT_TYPE" == "xPRETRAINED" ]; then
python train.py --config_path=$CONFIG_FILE --device_target="GPU" --data_path=$DATASET_PATH \
--pre_trained=$CKPT_FILE &> log &
else
python train.py --config_path=$CONFIG_FILE --device_target="GPU" --data_path=$DATASET_PATH &> log &
fi
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