未验证 提交 a288a3aa 编写于 作者: Z Zhang Ting 提交者: GitHub

fix det_v3 (#10180)

上级 2d44a71b
...@@ -18,6 +18,7 @@ Global: ...@@ -18,6 +18,7 @@ Global:
save_res_path: ./checkpoints/det_db/predicts_db.txt save_res_path: ./checkpoints/det_db/predicts_db.txt
distributed: true distributed: true
d2s_train_image_shape: [3, -1, -1] d2s_train_image_shape: [3, -1, -1]
amp_dtype: bfloat16
Architecture: Architecture:
name: DistillationModel name: DistillationModel
...@@ -222,4 +223,4 @@ Eval: ...@@ -222,4 +223,4 @@ Eval:
shuffle: False shuffle: False
drop_last: False drop_last: False
batch_size_per_card: 1 # must be 1 batch_size_per_card: 1 # must be 1
num_workers: 2 num_workers: 2
\ No newline at end of file
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册