andreas122001's picture
Upload folder using huggingface_hub
fa2ea9a verified
{
"add_after_accident_two_ways": -15,
"add_after_construction_obstacle_two_ways": 15,
"add_after_parked_obstacle_two_ways": -5,
"add_after_vehicle_opens_door_two_ways": -20,
"add_before_accident_two_ways": -15,
"add_before_construction_obstacle_two_ways": 15,
"add_before_parked_obstacle_two_ways": -5,
"add_before_vehicle_opens_door_two_ways": -20,
"add_features": 1,
"aim_distance_fast": 3.0,
"aim_distance_slow": 2.25,
"aim_distance_threshold": 5.5,
"angle_bins": [
-0.375,
-0.125,
0.125,
0.375
],
"angle_weights": [
204.25901201602136,
7.554315623148331,
0.21388916461734406,
5.476446162657503,
207.86684782608697
],
"angles": [
-0.5,
-0.25,
0.0,
0.25,
0.5
],
"attn_pdrop": 0.1,
"augment": 1,
"augment_percentage": 0.5,
"backbone": "transFuser",
"batch_size": 8,
"bb_confidence_threshold": 0.3,
"bb_input_channel": 64,
"bb_save_radius": 64.0,
"bev_classes_list": [
[
0,
0,
0
],
[
200,
200,
200
],
[
255,
255,
255
],
[
255,
255,
0
],
[
50,
234,
157
],
[
160,
160,
0
],
[
0,
255,
0
],
[
255,
255,
0
],
[
255,
0,
0
],
[
250,
170,
30
],
[
0,
255,
0
]
],
"bev_converter": [
0,
1,
2,
3,
4,
5,
6,
7,
8,
9,
10
],
"bev_down_sample_factor": 4,
"bev_features_chanels": 64,
"bev_grid_height_downsample_factor": 1,
"bev_latent_dim": 32,
"bev_semantic_weights": [
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0
],
"bev_upsample_factor": 2,
"bicycle_frame_rate": 20,
"block_exp": 4,
"brake_acceleration": -4.952399,
"brake_ratio": 1.1,
"brake_speed": 0.4,
"brake_uncertainty_threshold": 0.9,
"brake_values": {
"dtype": "float64",
"py/object": "numpy.ndarray",
"values": [
0.0093171137,
0.0820967431,
-0.00283832427,
5.06587474e-05,
-4.90357228e-07,
2.44419284e-09,
-4.91381935e-12
]
},
"braking_distance_calculation_safety_distance": 10,
"camera_fov": 110,
"camera_height": 512,
"camera_pos": [
-1.5,
0.0,
2.0
],
"camera_rot_0": [
0.0,
0.0,
0.0
],
"camera_rotation_augmentation_max": 5.0,
"camera_rotation_augmentation_min": -5.0,
"camera_translation_augmentation_max": 1.0,
"camera_translation_augmentation_min": -1.0,
"camera_width": 1024,
"carla_fps": 20,
"carla_frame_rate": 0.05,
"center_net_max_pooling_kernel": 3,
"check_path_free_safety_distance": 10,
"check_path_free_safety_time": 0.2,
"check_previous_distance_for_lane_change": 150,
"classes": {
"0": [
0,
0,
0
],
"1": [
30,
170,
250
],
"2": [
200,
200,
200
],
"3": [
255,
255,
0
],
"4": [
0,
255,
0
],
"5": [
0,
255,
255
],
"6": [
255,
255,
255
]
},
"classes_list": [
[
0,
0,
0
],
[
250,
170,
30
],
[
200,
200,
200
],
[
0,
255,
255
],
[
0,
255,
0
],
[
255,
255,
0
],
[
255,
255,
255
]
],
"clearing_distance_to_stop_sign": 3.0,
"clip_delta": 1.0,
"clip_throttle": 1.0,
"color_aug_prob": 0.5,
"compile": 0,
"compile_mode": "default",
"compute_min_time_to_cover_distance_params": {
"dtype": "float64",
"py/object": "numpy.ndarray",
"values": [
0.00904221,
0.00733342,
-0.03744807,
0.0235038
]
},
"continue_epoch": 0,
"converter": [
0,
2,
6,
0,
0,
0,
0,
3,
0,
0,
0,
0,
4,
0,
1,
1,
1,
0,
1,
1,
0,
0,
0,
0,
5,
0,
0,
0,
0
],
"cosine_t0": 1,
"cosine_t_mult": 2,
"cpu_cores": 32,
"creep_duration": 20,
"creep_throttle": 0.4,
"crop_bev": false,
"crop_bev_height_only_from_behind": 0,
"crop_image": 1,
"cropped_height": 384,
"cropped_width": 1024,
"data_roots": [
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/HazardAtSideLane",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/BlockedIntersection",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/ParkingCrossingPedestrian",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/SignalizedJunctionLeftTurn",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/HighwayExit",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/ParkingExit",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/ConstructionObstacleTwoWays",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/VehicleTurningRoutePedestrian",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/VehicleOpensDoorTwoWays",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/MergerIntoSlowTrafficV2",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/PriorityAtJunction",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/AccidentTwoWays",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/OppositeVehicleRunningRedLight",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/ParkedObstacle",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/ControlLoss",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/NonSignalizedJunctionLeftTurn",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/ConstructionObstacle",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/VehicleTurningRoute",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/HazardAtSideLaneTwoWays",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/ParkedObstacleTwoWays",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/Accident",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/noScenarios",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/PedestrianCrossing",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/HighwayCutIn",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/EnterActorFlowV2",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/ParkingCutIn",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/NonSignalizedJunctionRightTurn",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/CrossingBicycleFlow",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/HardBreakRoute",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/InvadingTurn",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/SignalizedJunctionRightTurn",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/MergerIntoSlowTraffic",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/InterurbanActorFlow",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/StaticCutIn",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/DynamicObjectCrossing",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/OppositeVehicleTakingPriority",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/InterurbanAdvancedActorFlow",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/YieldToEmergencyVehicle",
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data/EnterActorFlow"
],
"data_save_freq": 5,
"dataset_cache_name": "dataset_cache",
"debug": 0,
"deconv_channel_num_0": 128,
"deconv_channel_num_1": 64,
"deconv_channel_num_2": 32,
"deconv_scale_factor_0": 4,
"deconv_scale_factor_1": 8,
"default_forecast_length": 2.0,
"default_max_distance_to_process_scenario": 50,
"default_overtake_speed": 13.88888888888889,
"dense_route_planner_min_distance": 2.4,
"detailed_loss_weights": {
"loss_bev_semantic": 0.08333333333333333,
"loss_brake": 0.0,
"loss_center_heatmap": 0.08333333333333333,
"loss_checkpoint": 0.08333333333333333,
"loss_depth": 0.08333333333333333,
"loss_forcast": 0.0,
"loss_offset": 0.08333333333333333,
"loss_selection": 0.0,
"loss_semantic": 0.08333333333333333,
"loss_target_speed": 0.08333333333333333,
"loss_trajectories": 0.08333333333333333,
"loss_trajectory_confidence": 0.08333333333333333,
"loss_velocity": 0.0,
"loss_wh": 0.08333333333333333,
"loss_wp": 0.0,
"loss_yaw_class": 0.08333333333333333,
"loss_yaw_res": 0.08333333333333333
},
"detect_boxes": 1,
"detection_radius": 50.0,
"distance_to_delete_scenario_in_two_ways": 20,
"draw_future_route_till_distance": 500,
"draw_life_time": 0.051,
"ego_extent_x": 2.4508416652679443,
"ego_extent_y": 1.0641621351242065,
"ego_extent_z": 0.7553732395172119,
"ego_vehicles_route_point_search_distance": 40,
"embd_pdrop": 0.1,
"epochs": 31,
"estimate_class_distributions": 0,
"estimate_semantic_distribution": 0,
"extent_ego_bbs_speed_threshold": 5,
"extent_other_vehicles_bbs_speed_threshold": 1.0,
"extra_route_length": 50,
"extra_sensor_channels": 128,
"factor_accident_two_ways": 1.0,
"factor_construction_obstacle_two_ways": 1.0,
"factor_parked_obstacle_two_ways": 0.6,
"factor_vehicle_opens_door_two_ways": 0.475,
"fence_avoidance_margin_invading_turn": 0.3,
"focal_loss_gamma": 2.0,
"forcast_time": 0.5,
"forecast_length_lane_change": 1.1,
"fps": 20.0,
"fps_inv": 0.05,
"freeze_backbone": 0,
"front_wheel_base": -0.090769015,
"gpt_layer_norm_init_weight": 1.0,
"gpt_linear_layer_init_mean": 0.0,
"gpt_linear_layer_init_std": 0.02,
"grad_clip_max_norm": 1.0,
"gru_hidden_size": 64,
"gru_input_size": 256,
"high_speed_extent_factor_ego_x": 1.3,
"high_speed_extent_factor_ego_y": 1.2,
"high_speed_extent_x_factor_other_vehicle": 1.5,
"high_speed_extent_y_factor_other_vehicle": 1.3,
"high_speed_min_extent_x_other_vehicle": 1.2,
"high_speed_min_extent_x_other_vehicle_lane_change": 2.0,
"high_speed_min_extent_y_other_vehicle": 1.0,
"hist_max_per_pixel": 5,
"id": "stage2/tfpp_default",
"idm_acceleration_exponent": 4.0,
"idm_bicycle_desired_time_headway": 0.25,
"idm_bicycle_minimum_distance": 4.0,
"idm_comfortable_braking_deceleration_high_speed": 3.72,
"idm_comfortable_braking_deceleration_low_speed": 8.7,
"idm_comfortable_braking_deceleration_threshold": 6.02,
"idm_leading_vehicle_minimum_distance": 4.0,
"idm_leading_vehicle_time_headway": 0.25,
"idm_maximum_acceleration": 24.0,
"idm_pedestrian_desired_time_headway": 0.1,
"idm_pedestrian_minimum_distance": 4.0,
"idm_red_light_desired_time_headway": 0.1,
"idm_red_light_minimum_distance": 6.0,
"idm_stop_sign_desired_time_headway": 0.1,
"idm_stop_sign_minimum_distance": 2.0,
"idm_t_bound": 0.05,
"idm_two_way_scenarios_minimum_distance": 2.0,
"idm_two_way_scenarios_time_headway": 0.1,
"ignore_index": -999,
"image_architecture": "regnety_032",
"image_u_net_output_features": 512,
"img_horz_anchors": 32,
"img_seq_len": 1,
"img_step_size": 1,
"img_vert_anchors": 12,
"inference_direct_controller": true,
"inital_frames_delay": 40.0,
"input_path_to_target_speed_network": 0,
"iou_treshold_nms": 0.2,
"label_smoothing_alpha": 0.1,
"lane_shift_extension_length_for_yield_to_emergency_vehicle": 200,
"lateral_default_lookahead": 24,
"lateral_k_d": 1.3782508892109167,
"lateral_k_i": 0.6406067986034124,
"lateral_k_p": 3.118357247806046,
"lateral_n": 6,
"lateral_pid_default_lookahead": 24.0,
"lateral_pid_kd": 1.3782508892109167,
"lateral_pid_ki": 0.6406067986034124,
"lateral_pid_kp": 3.118357247806046,
"lateral_pid_maximum_lookahead_distance": 105.0,
"lateral_pid_minimum_lookahead_distance": 24.0,
"lateral_pid_speed_offset": 1.9152884533402488,
"lateral_pid_speed_scale": 0.9755321901954155,
"lateral_pid_speed_threshold": 23.150102938235136,
"lateral_pid_window_size": 6,
"lateral_speed_offset": 1.9152884533402488,
"lateral_speed_scale": 0.9755321901954155,
"lateral_speed_threshold": 23.150102938235136,
"leading_vehicles_max_route_angle_distance": 35.0,
"leading_vehicles_max_route_distance": 2.5,
"leading_vehicles_maximum_detection_radius": 800,
"learn_multi_task_weights": 0,
"learn_origin": 1,
"lidar_architecture": "regnety_032",
"lidar_aug_prob": 1.0,
"lidar_horz_anchors": 8,
"lidar_points_per_second": 600000,
"lidar_pos": [
0.0,
0.0,
2.5
],
"lidar_resolution_height": 256,
"lidar_resolution_width": 256,
"lidar_rot": [
0.0,
0.0,
-90.0
],
"lidar_rotation_frequency": 10,
"lidar_seq_len": 1,
"lidar_split_height": 0.2,
"lidar_step_size": 1,
"lidar_vert_anchors": 8,
"light_radius": 64.0,
"load_file": "/cluster/work/andrebw/repos/temporal_garage/results/training/tfpp_base/model_0030.pth",
"local_rank": -999,
"log_route_planner_min_distance": 4.0,
"logdir": "/cluster/work/andrebw/repos/temporal_garage/results/training/stage2/tfpp_default",
"logger_region_of_interest": 30.0,
"logging_freq": 10,
"longitudinal_linear_regression_maximum_acceleration": 1.89,
"longitudinal_linear_regression_maximum_deceleration": -4.82,
"longitudinal_linear_regression_minimum_target_speed": 0.278,
"longitudinal_linear_regression_params": {
"dtype": "float64",
"py/object": "numpy.ndarray",
"values": [
1.1990342347353184,
-0.8057602384167799,
1.710818710950062,
0.921890257450335,
1.556497522998393,
-0.7013479734904027,
1.031266635497984
]
},
"longitudinal_max_acceleration": 1.89,
"longitudinal_params": {
"py/tuple": [
1.1990342347353184,
-0.8057602384167799,
1.710818710950062,
0.921890257450335,
1.556497522998393,
-0.7013479734904027,
1.031266635497984
]
},
"longitudinal_pid_braking_ratio": 1.0324622059220139,
"longitudinal_pid_derivative_gain": 1.5761818624794222,
"longitudinal_pid_integral_gain": 0.2941563856687906,
"longitudinal_pid_max_window_length": 0,
"longitudinal_pid_minimum_target_speed": 0.278,
"longitudinal_pid_proportional_gain": 1.0016429066823955,
"longitudinal_pid_speed_error_scaling": 0.0,
"lr": 0.0003,
"max_blocked_ticks": 170,
"max_distance_lane_change_trailing_vehicles": 150,
"max_distance_to_overtake_two_way_scnearios": 80,
"max_distance_to_process_hazard_at_side_lane": 25,
"max_distance_to_process_hazard_at_side_lane_two_ways": 10,
"max_height_lidar": 100.0,
"max_lookahead_to_check_for_junction": 300,
"max_num_bbs": 30,
"max_num_trajectories": 20,
"max_speed_in_junction": 17.77777777777778,
"max_x": 32,
"max_y": 32,
"max_z": 4,
"max_z_projection": 14,
"min_walker_speed": 0.5,
"min_x": -32,
"min_y": -32,
"min_z": -4,
"min_z_projection": -10,
"minimum_lane_width_threshold": 2.5,
"minimum_lookahead_distance_to_compute_near_lane_change": 200,
"minimum_speed_to_prevent_rolling_back": 0.5,
"multi_step_lr_decay": 0.1,
"multi_wp_output": 0,
"n_head": 4,
"n_layer": 2,
"normalize_imagenet": true,
"num_bb_classes": 5,
"num_bev_semantic_classes": 11,
"num_decoder_heads": 8,
"num_dir_bins": 12,
"num_lidar_hits_for_detection_car": 1,
"num_lidar_hits_for_detection_walker": 1,
"num_repetitions": 1,
"num_route_points": 20,
"num_route_points_saved": 20,
"num_semantic_classes": 7,
"num_transformer_decoder_layers": 6,
"overtake_speed_vehicle_opens_door_two_ways": 11.11111111111111,
"pedestrian_minimum_extent": 1.5,
"perspective_downsample_factor": 1,
"pixels_per_meter": 4.0,
"pixels_per_meter_collection": 2.0,
"plant_embd_pdrop": 0.1,
"plant_hf_checkpoint": "prajjwal1/bert-medium",
"plant_max_speed_pred": 60.0,
"plant_num_attributes": 7,
"plant_object_types": 6,
"plant_precision_angle": 4,
"plant_precision_brake": 2,
"plant_precision_pos": 7,
"plant_precision_speed": 5,
"plant_pretraining": null,
"point_format": 0,
"point_precision": 0.01,
"points_per_meter": 10,
"pred_len": 8,
"predict_checkpoint_len": 10,
"previous_road_lane_retrieve_distance": 100,
"py/object": "config.GlobalConfig",
"ratio_target_speed_limit": 0.72,
"realign_lidar": 1,
"rear_wheel_base": 1.4178275,
"resid_pdrop": 0.1,
"return_temporal_attn_weights": false,
"root_dir": [
"/cluster/work/andrebw/repos/temporal_garage/results/data/garage_v2/data"
],
"route_planner_max_distance": 50.0,
"route_planner_min_distance": 7.5,
"route_points": 10,
"route_shift_end_distance_invading_turn": 100,
"route_shift_start_distance_invading_turn": 150,
"safety_box_x_max": 4.950841665267944,
"safety_box_x_min": 2.4508416652679443,
"safety_box_y_max": 0.8513297080993653,
"safety_box_y_min": -0.8513297080993653,
"safety_box_z_max": 1.5,
"safety_box_z_min": 0.5,
"schedule_reduce_epoch_01": 30,
"schedule_reduce_epoch_02": 40,
"seed": 0,
"semantic_weights": [
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0
],
"seq_len": 1,
"seq_step": 1,
"setting": "13_withheld",
"skip_first": 10,
"slow_speed_extent_factor_ego": 1.0,
"slower_factor": 0.8,
"smooth_route": 1,
"speed_kd": 2.0,
"speed_ki": 1.0,
"speed_kp": 1.75,
"speed_limit_waypoints_spacing_check": 50,
"speed_n": 20,
"steer_noise": 0.001,
"steering_gain": 0.36848336,
"stuck_threshold": 1100,
"sync_batch_norm": 0,
"tailing_vehicles_maximum_detection_radius": 800,
"target_speed_bins": [
4.001,
8.001,
10.001,
13.889888879999999,
16.001,
17.77877777,
20.001
],
"target_speed_weights": [
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0
],
"target_speeds": [
0.0,
4.0,
8.0,
10,
13.88888888,
16,
17.77777777,
20
],
"temporal_fusion_heads": 4,
"temporal_fusion_layers": 1,
"temporal_hidden_dim": 512,
"tf_first_checkpoint_distance": 25,
"throttle_acceleration": 0.5633837,
"throttle_threshold_during_forecasting": 0.3,
"throttle_values": {
"dtype": "float64",
"py/object": "numpy.ndarray",
"values": [
0.963873001,
0.000437535692,
-0.380192912,
1.74950069,
0.0916787414,
-0.070546153,
-0.00105996152,
0.000671079346
]
},
"time_step": 0.05,
"top_k_center_keypoints": 100,
"tp_attention": 0,
"trailing_vehicles_max_route_distance": 3.0,
"trailing_vehicles_max_route_distance_lane_change": 6.0,
"train_debug_save_freq": 1,
"train_sampling_rate": 1,
"trajectory_max_x": 310,
"trajectory_max_y": 380,
"trajectory_min_x": -70,
"trajectory_min_y": -120,
"trajectory_pred_len": 10,
"trajectory_step_size": 1,
"transformer_decoder_join": 1,
"transition_length_accident_two_ways": 40,
"transition_length_construction_obstacle_two_ways": 40,
"transition_length_parked_obstacle_two_ways": 40,
"transition_length_vehicle_opens_door_two_ways": 40,
"transition_smoothness_distance": 80,
"transition_smoothness_factor_construction_obstacle": 105.0,
"turn_kd": 0.3,
"turn_ki": 0.75,
"turn_kp": 1.25,
"turn_n": 20,
"two_tp_input": false,
"unclearing_distance_to_stop_sign": 10,
"use_amp": 0,
"use_bev_semantic": 1,
"use_color_aug": 1,
"use_controller_input_prediction": 1,
"use_cosine_schedule": 1,
"use_cutout": 0,
"use_dataset_pruning": true,
"use_depth": 1,
"use_discrete_command": 1,
"use_disk_cache": 1,
"use_focal_loss": 0,
"use_grad_clip": 0,
"use_ground_plane": 0,
"use_label_smoothing": 0,
"use_learnable_historic_initialization": true,
"use_optim_groups": 0,
"use_plant": 0,
"use_recurrent_training": true,
"use_semantic": 1,
"use_speed_weights": 0,
"use_temporal_fusion": 0,
"use_tp": 1,
"use_trajectory_prediction": 0,
"use_twohot_target_speeds": true,
"use_velocity": 1,
"use_velocity_brake_head": false,
"use_wp_gru": 0,
"val_every": 10,
"val_towns": [
13
],
"validation": true,
"weight_decay": 0.01,
"wp_dilation": 1,
"zero_redundancy_optimizer": 1
}