Skip to content

GitLab

Unverified Commit a7365a0b authored by Double_V's avatar Double_V Committed by GitHub
Browse files

Merge pull request #7407 from LDOUBLEV/25_fix 

fix cml
parents a0c4649d 99a5d59a
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 0 deletions
configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml
View file @ a7365a0b
...@@ -197,6 +197,7 @@ Train: ...@@ -197,6 +197,7 @@ Train:
drop_last: false drop_last: false
batch_size_per_card: 8 batch_size_per_card: 8
num_workers: 4 num_workers: 4
Eval: Eval:
dataset: dataset:
name: SimpleDataSet name: SimpleDataSet
...@@ -217,3 +218,8 @@ Eval: ...@@ -217,3 +218,8 @@ Eval:
- ToCHWImage: - ToCHWImage:
- KeepKeys: - KeepKeys:
keep_keys: ['image', 'shape', 'polys', 'ignore_tags'] keep_keys: ['image', 'shape', 'polys', 'ignore_tags']
loader:
shuffle: False
drop_last: False
batch_size_per_card: 1 # must be 1
num_workers: 2
Markdown is supported
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