fix klx config for detr 2x8 (FlagOpen#368)

Co-authored-by: root <root@szzj-isa-ai-chip1.szzj.baidu.com>
clveryang · Dec 22, 2023 · 2b757f0 · 2b757f0
1 parent c770384
commit 2b757f0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/training/kunlunxin/detr-pytorch/config/config_R300x2x8.py b/training/kunlunxin/detr-pytorch/config/config_R300x2x8.py
@@ -1,4 +1,4 @@
 from config_common import *
 
-train_batch_size = 8
-eval_batch_size = 8
+train_batch_size = 4
+eval_batch_size = 4