Add new dataset and collate function

dinhanhx · Aug 17, 2021 · 3e8aac1 · 3e8aac1
1 parent 621e640
commit 3e8aac1
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/UNITER/data/__init__.py b/UNITER/data/__init__.py
@@ -12,7 +12,14 @@
                     nlvr2_triplet_collate, nlvr2_triplet_eval_collate)
 from .hm import (HMDataset, HMEvalDataset, HMTestDataset,
                  hm_collate, hm_eval_collate, hm_test_collate,
+
                  HMPairedDataset, HMPairedEvalDataset, HMPairedTestDataset,
-                 hm_paired_collate, hm_paired_eval_collate, hm_paired_test_collate
+                 hm_paired_collate, hm_paired_eval_collate, hm_paired_test_collate,
+
+                 HMTripleDataset, HMTripleEvalDataset, HMTripleTestDataset,
+                 hm_triple_collate, hm_triple_eval_collate, hm_triple_test_collate,
+
+                 HMQuadDataset, HMQuadEvalDataset, HMQuadTestDataset,
+                 hm_quad_collate, hm_quad_eval_collate, hm_quad_test_collate
                  )