Upload config.json

config.json

@@ -0,0 +1,253 @@
+{
+    "target_dataset": 1,
+    "num_available_cameras": 3,
+    "used_cameras": [
+        true,
+        true,
+        true
+    ],
+    "num_used_cameras": 3,
+    "pixels_per_meter": 4.0,
+    "min_x_meter": -32,
+    "max_x_meter": 64,
+    "min_y_meter": -40,
+    "max_y_meter": 40,
+    "lidar_width_pixel": 384,
+    "lidar_height_pixel": 320,
+    "lidar_width_meter": 96,
+    "lidar_height_meter": 80,
+    "visualize_dataset": false,
+    "visualize_failed_scenarios": false,
+    "load_bev_3rd_person_images": false,
+    "seed": 0,
+    "wandb_id": "733_scaled_regnety_010_postrain32_0_251025_182327",
+    "wandb_resume": "allow",
+    "wandb_project_name": "lead_posttrain",
+    "description": "733_scaled_regnety_010_postrain32_0_251025_182327",
+    "visualize_training": true,
+    "id": "733_scaled_regnety_010_postrain32_0_251025_182327",
+    "load_file": ".../lead_ckpt/tfv6/model_0030_1.pth",
+    "continue_failed_training": false,
+    "epoch_checkpoints_keep": [],
+    "use_training_session_cache": false,
+    "use_persistent_cache": false,
+    "force_rebuild_data_cache": false,
+    "carla_cache_path": [
+        "1152",
+        "384",
+        "-32",
+        "64",
+        "-40",
+        "40",
+        "True",
+        "True",
+        "True",
+        "True",
+        "False",
+        "10"
+    ],
+    "training_session_cache_path": "/tmp/cruser",
+    "carla_root": "data/carla_leaderboard2",
+    "carla_data": "data/carla_leaderboard2/data",
+    "logdir": "outputs/local_training/posttrain_m1_ep30_k60",
+    "training_png_compression_level": 6,
+    "vehicle_min_num_lidar_points": 1,
+    "vehicle_min_num_visible_pixels": 1,
+    "pedestrian_min_num_lidar_points": 5,
+    "pedestrian_min_num_visible_pixels": 15,
+    "parking_vehicle_min_num_lidar_points": 3,
+    "parking_vehicle_min_num_visible_pixels": 5,
+    "grad_scaler_init_scale": 1024,
+    "grad_scaler_growth_factor": 2,
+    "grad_scaler_backoff_factor": 0.5,
+    "grad_scaler_growth_interval": 256,
+    "grad_scaler_max_grad_scale": 65536,
+    "plan_anchor_path": ".../lead_anchor/lead_cp_kmeans_60_10.npy",
+    "trajectory_sampling_num_poses": 10,
+    "tf_d_ffn": 1024,
+    "tf_d_model": 256,
+    "tf_num_head": 8,
+    "tf_dropout": 0.0,
+    "lidar_min_x": -32,
+    "lidar_max_x": 32,
+    "lidar_min_y": -32,
+    "lidar_max_y": 32,
+    "predict_ddbm": true,
+    "ddbm_training": true,
+    "sync_batchnorm": false,
+    "epochs": 31,
+    "batch_size": 64,
+    "use_mixed_precision_training": true,
+    "need_grad_scaler": false,
+    "use_zero_redundancy": false,
+    "save_model_checkpoint": true,
+    "is_pretraining": false,
+    "bucket_cap_mb": 256,
+    "prefetch_factor": 8,
+    "compile": true,
+    "channel_last": true,
+    "lr": 0.0003,
+    "skip_first": 8,
+    "skip_last": 8,
+    "radar_detection": true,
+    "use_radar_detection": true,
+    "num_radar_points_per_sensor": 75,
+    "num_radar_queries": 20,
+    "radar_hidden_dim_tokenizer": 1024,
+    "radar_token_dim": 256,
+    "radar_tf_dim_ff": 1024,
+    "radar_dropout": 0.1,
+    "radar_num_heads": 8,
+    "radar_num_layers": 4,
+    "radar_hidden_dim_decoder": 1024,
+    "radar_classification_loss_weight": 1.0,
+    "radar_regression_loss_weight": 5.0,
+    "num_radar_sensors": 4,
+    "force_rebuild_bucket": false,
+    "randomize_route_order": false,
+    "hold_out_town13_routes": false,
+    "bucket_collection_path": "data/carla_leaderboard2/buckets",
+    "use_sensor_perburtation": true,
+    "use_sensor_perburtation_prob": 0.5,
+    "use_color_aug": true,
+    "use_color_aug_prob": 0.2,
+    "weight_decay": 0.01,
+    "use_cosine_annealing_with_restarts": true,
+    "use_depth": true,
+    "training_used_lidar_steps": 10,
+    "min_z": -4,
+    "max_z": 4,
+    "hist_max_per_pixel": 5,
+    "lidar_vert_anchors": 10,
+    "lidar_horz_anchors": 12,
+    "detect_boxes": true,
+    "data_bb_static_types_white_list": [
+        "static.prop.constructioncone",
+        "static.prop.trafficwarning"
+    ],
+    "bb_confidence_threshold": 0.3,
+    "max_num_bbs": 90,
+    "num_dir_bins": 12,
+    "top_k_center_keypoints": 100,
+    "center_net_max_pooling_kernel": 3,
+    "bb_input_channel": 64,
+    "car_open_door_extra_width": 1.2,
+    "num_bb_classes": 8,
+    "use_discrete_command": true,
+    "discrete_command_dim": 6,
+    "use_noisy_tp": false,
+    "use_kalman_filter_for_gps": true,
+    "use_velocity": true,
+    "max_speed": 25.0,
+    "use_acceleration": false,
+    "max_acceleration": 10.0,
+    "use_previous_tp": true,
+    "use_next_tp": true,
+    "use_tp": true,
+    "target_points_normalization_constants": [
+        [
+            200.0,
+            50.0
+        ]
+    ],
+    "tp_pop_distance": 3.25,
+    "use_past_positions": false,
+    "use_past_speeds": false,
+    "num_past_samples_used": 0,
+    "transfuser_num_bev_cross_attention_layers": 6,
+    "transfuser_num_bev_cross_attention_heads": 8,
+    "transfuser_token_dim": 256,
+    "predict_target_speed": true,
+    "predict_spatial_path": true,
+    "predict_temporal_spatial_waypoints": true,
+    "use_planning_decoder": true,
+    "use_tfv5_planning_decoder": false,
+    "gru_hidden_size": 64,
+    "target_speed_classes": [
+        0.0,
+        4.0,
+        8.0,
+        10.0,
+        13.88888888,
+        16.0,
+        17.77777777,
+        20.0
+    ],
+    "target_speeds": [
+        0.0,
+        4.0,
+        8.0,
+        10.0,
+        13.88888888,
+        16.0,
+        17.77777777,
+        20.0
+    ],
+    "smooth_route": true,
+    "num_route_points_smoothing": 20,
+    "num_route_points_prediction": 10,
+    "max_distance_future_waypoint": 10.0,
+    "num_way_points_prediction": 8,
+    "waypoints_spacing": 5,
+    "horizontal_fov_reduction": 0,
+    "crop_height": 0,
+    "carla_crop_height_type": 3,
+    "image_width_before_camera_subselection": 1152,
+    "final_image_width": 1152,
+    "final_image_height": 384,
+    "img_vert_anchors": 12,
+    "img_horz_anchors": 36,
+    "freeze_backbone": false,
+    "image_architecture": "regnety_032",
+    "lidar_architecture": "regnety_032",
+    "LTF": false,
+    "block_exp": 4,
+    "n_layer": 2,
+    "n_head": 4,
+    "embd_pdrop": 0.1,
+    "resid_pdrop": 0.1,
+    "attn_pdrop": 0.1,
+    "gpt_linear_layer_init_mean": 0.0,
+    "gpt_linear_layer_init_std": 0.02,
+    "gpt_layer_norm_init_weight": 1.0,
+    "use_semantic": true,
+    "num_semantic_classes": 10,
+    "perspective_downsample_factor": 1,
+    "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,
+    "use_bev_semantic": true,
+    "num_bev_semantic_classes": 13,
+    "scale_pedestrian_bev_semantic_size": 2.5,
+    "pedestrian_bev_min_extent": 0.4,
+    "bev_features_chanels": 64,
+    "bev_down_sample_factor": 4,
+    "bev_upsample_factor": 2,
+    "use_carla_data": true,
+    "carla_num_samples": -1,
+    "use_navsim_data": false,
+    "navsim_data_root": "data/navsim_training_cache/trainval",
+    "navsim_num_samples": -1,
+    "schedule_carla_num_samples": false,
+    "use_waymo_e2e_data": false,
+    "waymo_e2e_num_training_samples": -1,
+    "waymo_e2e_training_data_root": "data/waymo_open_dataset_end_to_end_camera_v_1_0_0_training",
+    "waymo_e2e_val_data_root": "data/waymo_open_dataset_end_to_end_camera_v_1_0_0_val_rfm",
+    "waymo_e2e_test_data_root": "data/waymo_open_dataset_end_to_end_camera_v_1_0_0_test_submission",
+    "waymo_e2e_subsample_factor": 5,
+    "navsim_num_bev_semantic_classes": 7,
+    "navsim_num_bb_classes": 1,
+    "mixed_data_training": false,
+    "carla_leaderboard_mode": true,
+    "log_scalars_frequency": 1,
+    "log_images_frequency": 100,
+    "log_wandb": false,
+    "gpu_name": "h20",
+    "rank": 0,
+    "world_size": 2,
+    "local_rank": 0,
+    "assigned_cpu_cores": 8,
+    "workers_per_cpu_cores": 1
+}