Skip to content

Commit 33bf4da

Browse files
authored
[TIPC] Smaller batch size to avoid OOM, remove url annotation (PaddlePaddle#1665)
1 parent 328ae31 commit 33bf4da

File tree

4 files changed

+3
-5
lines changed

4 files changed

+3
-5
lines changed

test_tipc/configs/bisenetv2/train_infer_python.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ Global.use_gpu:null|null
66
Global.auto_cast:null
77
--iters:lite_train_lite_infer=50|lite_train_whole_infer=50|whole_train_whole_infer=1000
88
--save_dir:
9-
--batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=8
9+
--batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=3
1010
Global.pretrained_model:null
1111
train_model_name:best_model/model.pdparams
1212
train_infer_img_dir:test_tipc/data/cityscapes/val.list

test_tipc/configs/ocrnet_hrnetw18/train_infer_python.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ Global.use_gpu:null|null
66
Global.auto_cast:null
77
--iters:lite_train_lite_infer=50|lite_train_whole_infer=50|whole_train_whole_infer=1000
88
--save_dir:
9-
--batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=8
9+
--batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=3
1010
Global.pretrained_model:null
1111
train_model_name:best_model/model.pdparams
1212
train_infer_img_dir:test_tipc/data/cityscapes/val.list

test_tipc/configs/segformer_b0/train_infer_python.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ Global.use_gpu:null|null
66
Global.auto_cast:null
77
--iters:lite_train_lite_infer=50|lite_train_whole_infer=50|whole_train_whole_infer=1000
88
--save_dir:
9-
--batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=8
9+
--batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=3
1010
Global.pretrained_model:null
1111
train_model_name:best_model/model.pdparams
1212
train_infer_img_dir:test_tipc/data/cityscapes/val.list

test_tipc/docs/test_train_inference_python.md

-2
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,6 @@
22

33
Linux端基础训练预测功能测试的主程序为`test_train_inference_python.sh`,可以测试基于Python的模型训练、评估、推理等基本功能,包括裁剪、量化、蒸馏。
44

5-
<!-- - Mac端基础训练预测功能测试参考[链接](./mac_test_train_inference_python.md)
6-
- Windows端基础训练预测功能测试参考[链接](./win_test_train_inference_python.md) -->
75

86
## 1. 测试结论汇总
97

0 commit comments

Comments
 (0)