Jiangtao Hu
Jiangtao Hu
``` SELECT CASE WHEN tl.status == "green" THEN 0 WHEN tl.status == "yellow" THEN 1 WHEN tl.status == "red" THEN 2 ELSE 3 END AS status, tl.lane_connector_id, lp.timestamp AS timestamp...
``` # ego height nuplan/common/actor_state/vehicle_parameters.py def get_pacifica_parameters() -> VehicleParameters: """ :return VehicleParameters containing parameters of Pacifica Vehicle. """ return VehicleParameters( vehicle_name="pacifica", vehicle_type="gen1", width=1.1485 * 2.0, front_length=4.049, rear_length=1.127, wheel_base=3.089, cog_position_from_rear_axle=1.67, height=1.777,...
``` from deepspeed.utils.zero_to_fp32 import get_fp32_state_dict_from_zero_checkpoint # remove all the deepspeed magic data state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir) model = model.cpu() # move to cpu model.load_state_dict(state_dict) MODEL_PATH = "your_model_without_DS.pth" # Save the model's...
zero 0 or zero 1, the model parameters are not partitioned. It means the full FP32 state dictionary is readily accessible without additional reconstruction steps.
I tried the latest version(0.6.3.dev148+gca77dd7a.d20241009) and it works in chrome and Edge browser.