Merge branch 'local' into main

OpenRobotLab · Apr 24, 2024 · 0ee6e61 · 0ee6e61
2 parents e144c4b + 1f3743f
commit 0ee6e61
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 10 deletions.
diff --git a/...8xb12_embodiedscan-vg-9dof_complex-all.py → ..._8xb6_embodiedscan-vg-9dof_complex-all.py b/...8xb12_embodiedscan-vg-9dof_complex-all.py → ..._8xb6_embodiedscan-vg-9dof_complex-all.py
@@ -133,8 +133,8 @@
 
 # TODO: to determine a reasonable batch size
 train_dataloader = dict(
-    batch_size=12,
-    num_workers=12,
+    batch_size=6,
+    num_workers=6,
     persistent_workers=True,
     sampler=dict(type='DefaultSampler', shuffle=True),
     dataset=dict(type='RepeatDataset',
@@ -149,8 +149,8 @@
                               filter_empty_gt=True,
                               box_type_3d='Euler-Depth')))
 
-val_dataloader = dict(batch_size=12,
-                      num_workers=12,
+val_dataloader = dict(batch_size=6,
+                      num_workers=6,
                       persistent_workers=True,
                       drop_last=False,
                       sampler=dict(type='DefaultSampler', shuffle=False),
@@ -164,8 +164,8 @@
                                    filter_empty_gt=True,
                                    box_type_3d='Euler-Depth'))
 
-test_dataloader = dict(batch_size=12,
-                       num_workers=12,
+test_dataloader = dict(batch_size=6,
+                       num_workers=6,
                        persistent_workers=True,
                        drop_last=False,
                        sampler=dict(type='DefaultSampler', shuffle=False),

diff --git a/data/README.md b/data/README.md
@@ -28,10 +28,14 @@ data
 ├── embodiedscan_occupancy
 ├── embodiedscan_infos_train.pkl
 ├── embodiedscan_infos_val.pkl
-├── embodiedscan_infos_train_full_vg.json
-├── embodiedscan_infos_val_full_vg.json
-├── embodiedscan_infos_train_mini_vg.json
-├── embodiedscan_infos_val_mini_vg.json
+├── embodiedscan_infos_test.pkl
+├── embodiedscan_infos_train_vg.json
+├── embodiedscan_infos_val_vg.json
+├── embodiedscan_infos_test_vg.json
+├── embodiedscan_infos_train_mini_vg.json (mini set)
+├── embodiedscan_infos_val_mini_vg.json (mini set)
+├── embodiedscan_infos_train_vg_all.json (w/ complex prompts)
+├── embodiedscan_infos_val_vg_all.json (w/ complex prompts)
 ```
 
 5. Enter the project root directory, extract images by running