提交 99a5d59a 编写于 作者: L LDOUBLEV

fix cml

上级 97cb2d37
...@@ -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% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册