remove self.seq_name_to_infos of WOD if we do not use multi-frame tra…

…ining
iamwilliamli · Sep 2, 2022 · f56e4f1 · f56e4f1
1 parent 72d587d
commit f56e4f1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pcdet/datasets/waymo/waymo_dataset.py b/pcdet/datasets/waymo/waymo_dataset.py
@@ -78,7 +78,10 @@ def include_waymo_data(self, mode):
                 sampled_waymo_infos.append(self.infos[k])
             self.infos = sampled_waymo_infos
             self.logger.info('Total sampled samples for Waymo dataset: %d' % len(self.infos))
-
+
+        use_sequence_data = self.dataset_cfg.get('SEQUENCE_CONFIG', None) is not None and self.dataset_cfg.SEQUENCE_CONFIG.ENABLED
+        if not use_sequence_data:
+            seq_name_to_infos = None 
         return seq_name_to_infos
 
     def load_data_to_shared_memory(self):