Skip to content

Commit fcfb558

Browse files
author
yue kun
committed
adding ctdet mid for lore
1 parent 8b65d5a commit fcfb558

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

DocumentUnderstanding/LORE-TSR/src/lib/trains/train_factory.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,5 +6,6 @@
66

77
train_factory = {
88
'ctdet': CtdetTrainer,
9+
'ctdet_mid': CtdetTrainer,
910
'ctdet_small': CtdetTrainer
1011
}

0 commit comments

Comments
 (0)