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, +)