From b0fa4c5792f51226c04ba8fea0e1609fb12132de Mon Sep 17 00:00:00 2001 From: siddhi47 Date: Mon, 4 Aug 2025 16:44:18 -0700 Subject: [PATCH] Added new version of dataset to config --- waymax/config.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/waymax/config.py b/waymax/config.py index 5877456..44dab1a 100644 --- a/waymax/config.py +++ b/waymax/config.py @@ -278,3 +278,20 @@ def __post_init__(self): max_num_rg_points=30000, data_format=DataFormat.TFRECORD, ) + +WOD_1_3_0_TRAINING = DatasetConfig( + path="gs://waymo_open_dataset_motion_v_1_3_0/uncompressed/tf_example/training/training_tfexample.tfrecord@1000", + max_num_rg_points=10000, + data_format=DataFormat.TFRECORD, +) + + +WOD_1_3_0_TEST = DatasetConfig( + path="gs://waymo_open_dataset_motion_v_1_3_0/uncompressed/tf_example/testing/testing_tfexample.tfrecord@150", + data_format=DataFormat.TFRECORD, +) + +WOD_1_3_0_VALIDATION = DatasetConfig( + path="gs://waymo_open_dataset_motion_v_1_3_0/uncompressed/tf_example/validation/validation_tfexample.tfrecord@150", + data_format=DataFormat.TFRECORD, +)