parameters.json

{
    "TRAIN_ZARR": "scenes/train.zarr",
    "VALID_ZARR": "scenes/validate.zarr",
    "HBACKWARD" : 15,
    "HFORWARD" : 50,
    "NFRAMES" : 10,
    "FRAME_STRIDE": 15,
    "AGENT_FEATURE_DIM": 8,
    "MAX_AGENTS": 150,

    "NUM_WORKERS": 4,
    "BATCH_SIZE": 115,
    "EPOCHS": 40,
    "GRADIENT_CLIP_VAL": 1.0,
    "LIMIT_VAL_BATCHES": 0.20,
    "ROOT": "pointnet",

    "PERCEPTION_LABELS": [
        "PERCEPTION_LABEL_NOT_SET",
        "PERCEPTION_LABEL_UNKNOWN",
        "PERCEPTION_LABEL_DONTCARE",
        "PERCEPTION_LABEL_CAR",
        "PERCEPTION_LABEL_VAN",
        "PERCEPTION_LABEL_TRAM",
        "PERCEPTION_LABEL_BUS",
        "PERCEPTION_LABEL_TRUCK",
        "PERCEPTION_LABEL_EMERGENCY_VEHICLE",
        "PERCEPTION_LABEL_OTHER_VEHICLE",
        "PERCEPTION_LABEL_BICYCLE",
        "PERCEPTION_LABEL_MOTORCYCLE",
        "PERCEPTION_LABEL_CYCLIST",
        "PERCEPTION_LABEL_MOTORCYCLIST",
        "PERCEPTION_LABEL_PEDESTRIAN",
        "PERCEPTION_LABEL_ANIMAL",
        "AVRESEARCH_LABEL_DONTCARE"
    ],
    "KEPT_PERCEPTION_LABELS": [
        "PERCEPTION_LABEL_UNKNOWN",
        "PERCEPTION_LABEL_CAR",
        "PERCEPTION_LABEL_CYCLIST",
        "PERCEPTION_LABEL_PEDESTRIAN"
    ],
    "LEARNING_RATE": 2e-3,
    "WEIGHT_DECAY": 1e-6
}