Skip to content

Commit

Permalink
add cinn converge case
Browse files Browse the repository at this point in the history
  • Loading branch information
@Zeref996
Zeref996 committed Feb 13, 2025
1 parent 665845d commit 93aed3f
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion framework/e2e/cinn_converge/PP-YOLOE_plus-S_CINN.sh
View file
Original file line number Diff line number Diff line change
@@ -1 +1 @@
python main.py -c paddlex/configs/object_detection/PP-YOLOE_plus-S.yaml -o Global.mode=train -o Global.dataset_dir=../coco -o Train.num_classes=80 -o Train.epochs_iters=80 -o Train.batch_size=8 -o Train.learning_rate=0.001 -o Global.device=gpu:0,1,2,3,4,5,6,7 -o Global.output='./output/object_detection/PP-YOLOE_plus-S_CINN' -o Train.dy2st=True
python main.py -c paddlex/configs/modules/object_detection/PP-YOLOE_plus-S.yaml -o Global.mode=train -o Global.dataset_dir=../coco -o Train.num_classes=80 -o Train.epochs_iters=80 -o Train.batch_size=8 -o Train.learning_rate=0.001 -o Global.device=gpu:0,1,2,3,4,5,6,7 -o Global.output='./output/object_detection/PP-YOLOE_plus-S_CINN' -o Train.dy2st=True
2 changes: 1 addition & 1 deletion framework/e2e/cinn_converge/PP-YOLOE_plus-S_dy.sh
View file
Original file line number Diff line number Diff line change
@@ -1 +1 @@
python main.py -c paddlex/configs/object_detection/PP-YOLOE_plus-S.yaml -o Global.mode=train -o Global.dataset_dir=../coco -o Train.num_classes=80 -o Train.epochs_iters=80 -o Train.batch_size=8 -o Train.learning_rate=0.001 -o Global.device=gpu:0,1,2,3,4,5,6,7 -o Global.output='./output/object_detection/PP-YOLOE_plus-S_dy'
python main.py -c paddlex/configs/modules/object_detection/PP-YOLOE_plus-S.yaml -o Global.mode=train -o Global.dataset_dir=../coco -o Train.num_classes=80 -o Train.epochs_iters=80 -o Train.batch_size=8 -o Train.learning_rate=0.001 -o Global.device=gpu:0,1,2,3,4,5,6,7 -o Global.output='./output/object_detection/PP-YOLOE_plus-S_dy'

0 comments on commit 93aed3f

Please sign in to comment.