fix ci and output

This commit is contained in:
LDOUBLEV 2021-07-05 03:51:14 +00:00
parent 65118f3099
commit 4042247c97
3 changed files with 8 additions and 9 deletions

View File

@ -1,7 +1,7 @@
model_name:ocr_det
python:python3.7
gpu_list:-1|0|0,1
Global.auto_cast:False|True
gpu_list:0
Global.auto_cast:False
Global.epoch_num:10
Global.save_model_dir:./output/
Global.save_inference_dir:./output/
@ -9,7 +9,7 @@ Train.loader.batch_size_per_card:
Global.use_gpu
Global.pretrained_model
trainer:norm|pact|fpgm
trainer:norm|pact
norm_train:tools/train.py -c configs/det/det_mv3_db.yml -o Global.pretrained_model=./pretrain_models/MobileNetV3_large_x0_5_pretrained
quant_train:deploy/slim/quantization/quant.py -c configs/det/det_mv3_db.yml -o Global.pretrained_model=./pretrain_models/det_mv3_db_v2.0_train/best_accuracy
fpgm_train:null

View File

@ -106,7 +106,7 @@ class TextDetector(object):
model_precision=args.precision,
batch_size=1,
data_shape="dynamic",
save_path="./output/auto_log.lpg",
save_path=args.save_log_path,
inference_config=self.config,
pids=pid,
process_name=None,
@ -265,4 +265,3 @@ if __name__ == "__main__":
if args.benchmark:
text_detector.autolog.report()