Skip to content

Commit e9a8025

Browse files
committed
fix a bug in train.sh
1 parent 719fcd7 commit e9a8025

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

train.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ elif [[ $MODEL == *"plain_detr"* ]]; then
2222
EVAL_EPOCH=2
2323
elif [[ $MODEL == *"rtdetr"* ]]; then
2424
# Epoch setting
25-
BATCH_SIZE=1
25+
BATCH_SIZE=16
2626
EVAL_EPOCH=2
2727
fi
2828

0 commit comments

Comments
 (0)