Replies: 4 comments
-
报错补充:LAUNCH INFO 2024-06-17 17:01:27,787 Pod failed |
Beta Was this translation helpful? Give feedback.
-
你进'/root/workspace/paddleocr_train/log'看看,报错日志应该在这里面 |
Beta Was this translation helpful? Give feedback.
-
日志没有报错,只是提示code -9 |
Beta Was this translation helpful? Give feedback.
-
code -9 就是异常退出了,真正的log你要cd到/root/workspace/paddleocr_train/log这个目录下去看 |
Beta Was this translation helpful? Give feedback.
Uh oh!
There was an error while loading. Please reload this page.
-
问题描述 / Problem Description
使用PP-OCRv3做文字检测微调,在T4上跑训练无法训练,报错返回-9
运行环境 / Runtime Environment
复现代码 / Reproduction Code
代码直接使用了官方最新代码,配置文件如下:
Global:
debug: false
use_gpu: true
epoch_num: 500
log_smooth_window: 20
print_batch_step: 10
save_model_dir: ./output/ch_PP-OCR_V3_det/
save_epoch_step: 50
eval_batch_step:
cal_metric_during_train: false
pretrained_model: https://paddleocr.bj.bcebos.com/pretrained/MobileNetV3_large_x0_5_pretrained.pdparams
checkpoints: null
save_inference_dir: null
use_visualdl: false
infer_img: doc/imgs_en/img_10.jpg
save_res_path: ./checkpoints/det_db/predicts_db.txt
distributed: true
Architecture:
model_type: det
algorithm: DB
Transform:
Backbone:
name: MobileNetV3
scale: 0.5
model_name: large
disable_se: True
Neck:
name: RSEFPN
out_channels: 96
shortcut: True
Head:
name: DBHead
k: 50
Loss:
name: DBLoss
balance_loss: true
main_loss_type: DiceLoss
alpha: 5
beta: 10
ohem_ratio: 3
Optimizer:
name: Adam
beta1: 0.9
beta2: 0.999
lr:
name: Cosine
learning_rate: 0.001
warmup_epoch: 2
regularizer:
name: L2
factor: 5.0e-05
PostProcess:
name: DBPostProcess
thresh: 0.3
box_thresh: 0.6
max_candidates: 1000
unclip_ratio: 1.5
Metric:
name: DetMetric
main_indicator: hmean
Train:
dataset:
name: SimpleDataSet
data_dir: ../dataset/handwriting_data_det/train/20240605/train/images/
label_file_list:
- ../dataset/handwriting_data_det/train/20240605/val.txt
ratio_list: [1.0]
transforms:
- DecodeImage:
img_mode: BGR
channel_first: false
- DetLabelEncode: null
- IaaAugment:
augmenter_args:
- type: Fliplr
args:
p: 0.5
- type: Affine
args:
rotate:
- -10
- 10
- type: Resize
args:
size:
- 0.5
- 3
- EastRandomCropData:
size:
- 960
- 960
max_tries: 50
keep_ratio: true
- MakeBorderMap:
shrink_ratio: 0.4
thresh_min: 0.3
thresh_max: 0.7
- MakeShrinkMap:
shrink_ratio: 0.4
min_text_size: 8
- NormalizeImage:
scale: 1./255.
mean:
- 0.485
- 0.456
- 0.406
std:
- 0.229
- 0.224
- 0.225
order: hwc
- ToCHWImage: null
- KeepKeys:
keep_keys:
- image
- threshold_map
- threshold_mask
- shrink_map
- shrink_mask
loader:
shuffle: true
drop_last: false
batch_size_per_card: 1
num_workers: 0
Eval:
dataset:
name: SimpleDataSet
data_dir: ../dataset/handwriting_data_det/train/20240605/val/images/
label_file_list:
- ../dataset/handwriting_data_det/train/20240605/val.txt
transforms:
- DecodeImage:
img_mode: BGR
channel_first: false
- DetLabelEncode: null
- DetResizeForTest: null
- NormalizeImage:
scale: 1./255.
mean:
- 0.485
- 0.456
- 0.406
std:
- 0.229
- 0.224
- 0.225
order: hwc
- ToCHWImage: null
- KeepKeys:
keep_keys:
- image
- shape
- polys
- ignore_tags
loader:
shuffle: false
drop_last: false
batch_size_per_card: 1
num_workers: 0
完整报错 / Complete Error Message
没有完整的报错,直接提示code -9
可能解决方案 / Possible solutions
目前通过调整num_workers = 0,batch_size_per_card=1 能正常跑几轮,但是效率较低
附件 / Appendix
硬件配置
Beta Was this translation helpful? Give feedback.
All reactions