The problem of model feature extraction from decoder

[pisces365]

Hello, thank you for your excellent work. In memfuser.py, how are 2500 and 2501:2506 determined in the features queried from decoder?

        traffic_feature = hs[:, :2500]
        traffic_light_state_feature = hs[:, 2500]
        stop_sign_feature = hs[:, 2500]
        waypoints_feature = hs[:, 2501:2506]

Thanks!