Long Nguyen
commited on
Upload config.json with huggingface_hub
Browse files- config.json +254 -0
config.json
ADDED
|
@@ -0,0 +1,254 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"target_dataset": 3,
|
| 3 |
+
"num_available_cameras": 4,
|
| 4 |
+
"used_cameras": [
|
| 5 |
+
true,
|
| 6 |
+
true,
|
| 7 |
+
true,
|
| 8 |
+
true
|
| 9 |
+
],
|
| 10 |
+
"num_used_cameras": 4,
|
| 11 |
+
"pixels_per_meter": 4.0,
|
| 12 |
+
"min_x_meter": -32,
|
| 13 |
+
"max_x_meter": 32,
|
| 14 |
+
"min_y_meter": -32,
|
| 15 |
+
"max_y_meter": 32,
|
| 16 |
+
"lidar_width_pixel": 256,
|
| 17 |
+
"lidar_height_pixel": 256,
|
| 18 |
+
"lidar_width_meter": 64,
|
| 19 |
+
"lidar_height_meter": 64,
|
| 20 |
+
"visualize_dataset": false,
|
| 21 |
+
"visualize_failed_scenarios": false,
|
| 22 |
+
"load_bev_3rd_person_images": false,
|
| 23 |
+
"seed": 0,
|
| 24 |
+
"wandb_id": "810_w_schedule_w_bn_010_postrain32_0_251112_003238",
|
| 25 |
+
"wandb_resume": "never",
|
| 26 |
+
"description": "810_w_schedule_w_bn_010_postrain32_0_251112_003238",
|
| 27 |
+
"visualize_training": true,
|
| 28 |
+
"id": "810_w_schedule_w_bn_010_postrain32_0_251112_003238",
|
| 29 |
+
"load_file": "outputs/training/810_w_schedule_w_bn/000_pretrain2_0/251111_160533/model_0030.pth",
|
| 30 |
+
"continue_epoch": false,
|
| 31 |
+
"epoch_checkpoints_keep": [
|
| 32 |
+
7,
|
| 33 |
+
15,
|
| 34 |
+
31,
|
| 35 |
+
63,
|
| 36 |
+
127,
|
| 37 |
+
255,
|
| 38 |
+
511
|
| 39 |
+
],
|
| 40 |
+
"use_training_session_cache": true,
|
| 41 |
+
"use_persistent_cache": false,
|
| 42 |
+
"force_rebuild_data_cache": false,
|
| 43 |
+
"carla_cache_path": [
|
| 44 |
+
"1920",
|
| 45 |
+
"270",
|
| 46 |
+
"-32",
|
| 47 |
+
"32",
|
| 48 |
+
"-32",
|
| 49 |
+
"32",
|
| 50 |
+
"True",
|
| 51 |
+
"False",
|
| 52 |
+
"True",
|
| 53 |
+
"True",
|
| 54 |
+
"False",
|
| 55 |
+
"10",
|
| 56 |
+
"8",
|
| 57 |
+
"5"
|
| 58 |
+
],
|
| 59 |
+
"ssd_cache_path": "/scratch/1909240",
|
| 60 |
+
"carla_root": "data/carla_leaderboad2_v12/results/data/sensor_data/",
|
| 61 |
+
"carla_data": "data/carla_leaderboad2_v12/results/data/sensor_data/data",
|
| 62 |
+
"logdir": "/home/nguyen/master_thesis/outputs/training/810_w_schedule_w_bn/010_postrain32_0/251112_003238",
|
| 63 |
+
"training_png_compression_level": 6,
|
| 64 |
+
"vehicle_min_num_lidar_points": 1,
|
| 65 |
+
"vehicle_min_num_visible_pixels": 1,
|
| 66 |
+
"pedestrian_min_num_lidar_points": 5,
|
| 67 |
+
"pedestrian_min_num_visible_pixels": 15,
|
| 68 |
+
"parking_vehicle_min_num_lidar_points": 3,
|
| 69 |
+
"parking_vehicle_min_num_visible_pixels": 5,
|
| 70 |
+
"grad_scaler_init_scale": 1024,
|
| 71 |
+
"grad_scaler_growth_factor": 2,
|
| 72 |
+
"grad_scaler_backoff_factor": 0.5,
|
| 73 |
+
"grad_scaler_growth_interval": 256,
|
| 74 |
+
"grad_scaler_max_grad_scale": 65536,
|
| 75 |
+
"sync_batchnorm": false,
|
| 76 |
+
"freeze_batchnorm": false,
|
| 77 |
+
"epochs": 61,
|
| 78 |
+
"batch_size": 64,
|
| 79 |
+
"use_mixed_precision_training": true,
|
| 80 |
+
"need_grad_scaler": false,
|
| 81 |
+
"use_zero_redundancy": false,
|
| 82 |
+
"save_model_checkpoint": true,
|
| 83 |
+
"is_pretraining": false,
|
| 84 |
+
"prefetch_factor": 8,
|
| 85 |
+
"compile": true,
|
| 86 |
+
"channel_last": true,
|
| 87 |
+
"lr": 0.0003,
|
| 88 |
+
"skip_first": 8,
|
| 89 |
+
"skip_last": 16,
|
| 90 |
+
"radar_detection": false,
|
| 91 |
+
"use_radar_detection": false,
|
| 92 |
+
"num_radar_points_per_sensor": 75,
|
| 93 |
+
"num_radar_queries": 20,
|
| 94 |
+
"radar_hidden_dim_tokenizer": 1024,
|
| 95 |
+
"radar_token_dim": 256,
|
| 96 |
+
"radar_tf_dim_ff": 1024,
|
| 97 |
+
"radar_dropout": 0.1,
|
| 98 |
+
"radar_num_heads": 8,
|
| 99 |
+
"radar_num_layers": 4,
|
| 100 |
+
"radar_hidden_dim_decoder": 1024,
|
| 101 |
+
"radar_classification_loss_weight": 1.0,
|
| 102 |
+
"radar_forecast_loss_weight": 0.1,
|
| 103 |
+
"radar_regression_loss_weight": 5.0,
|
| 104 |
+
"radar_sine_features": 128,
|
| 105 |
+
"num_radar_sensors": 4,
|
| 106 |
+
"forecast_radar_detections": false,
|
| 107 |
+
"num_radar_forecast_steps": 8,
|
| 108 |
+
"radar_waypoints_spacing": 5,
|
| 109 |
+
"max_distance_future_waypoint": 10.0,
|
| 110 |
+
"force_rebuild_bucket": false,
|
| 111 |
+
"randomize_route_order": false,
|
| 112 |
+
"subsample_data": false,
|
| 113 |
+
"hold_out_town13_routes": false,
|
| 114 |
+
"carla_bucket_collection": "full_posttrain",
|
| 115 |
+
"bucket_collection_path": "data/carla_leaderboad2_v12/results/data/sensor_data/buckets",
|
| 116 |
+
"use_sensor_perburtation": true,
|
| 117 |
+
"use_sensor_perburtation_prob": 0.8,
|
| 118 |
+
"use_extreme_sensor_perburtation_prob": 0.5,
|
| 119 |
+
"use_color_aug": true,
|
| 120 |
+
"use_color_aug_prob": 0.1,
|
| 121 |
+
"weight_decay": 0.01,
|
| 122 |
+
"use_grad_clip": false,
|
| 123 |
+
"grad_quantile": 0.99,
|
| 124 |
+
"grad_history_length": 1000,
|
| 125 |
+
"use_optimizer_groups": false,
|
| 126 |
+
"use_cosine_annealing_with_restarts": true,
|
| 127 |
+
"use_depth": false,
|
| 128 |
+
"training_used_lidar_steps": 10,
|
| 129 |
+
"min_z": -4,
|
| 130 |
+
"max_z": 4,
|
| 131 |
+
"hist_max_per_pixel": 5,
|
| 132 |
+
"lidar_vert_anchors": 8,
|
| 133 |
+
"lidar_horz_anchors": 8,
|
| 134 |
+
"detect_boxes": true,
|
| 135 |
+
"debug_boxes_visualization": true,
|
| 136 |
+
"center_net_global_avg_factor": false,
|
| 137 |
+
"data_bb_static_types_white_list": [
|
| 138 |
+
"static.prop.constructioncone",
|
| 139 |
+
"static.prop.trafficwarning"
|
| 140 |
+
],
|
| 141 |
+
"bb_confidence_threshold": 0.3,
|
| 142 |
+
"max_num_bbs": 90,
|
| 143 |
+
"num_dir_bins": 12,
|
| 144 |
+
"top_k_center_keypoints": 100,
|
| 145 |
+
"center_net_max_pooling_kernel": 3,
|
| 146 |
+
"bb_input_channel": 64,
|
| 147 |
+
"car_open_door_extra_width": 1.2,
|
| 148 |
+
"num_bb_classes": 8,
|
| 149 |
+
"use_discrete_command": true,
|
| 150 |
+
"discrete_command_dim": 4,
|
| 151 |
+
"use_noisy_tp": false,
|
| 152 |
+
"use_velocity": true,
|
| 153 |
+
"max_speed": 15.0,
|
| 154 |
+
"use_acceleration": true,
|
| 155 |
+
"max_acceleration": 4.0,
|
| 156 |
+
"use_previous_tp": false,
|
| 157 |
+
"use_next_tp": false,
|
| 158 |
+
"use_tp": false,
|
| 159 |
+
"target_points_normalization_constants": [
|
| 160 |
+
[
|
| 161 |
+
200.0,
|
| 162 |
+
50.0
|
| 163 |
+
]
|
| 164 |
+
],
|
| 165 |
+
"tp_pop_distance": 3.25,
|
| 166 |
+
"use_past_positions": false,
|
| 167 |
+
"use_past_speeds": false,
|
| 168 |
+
"num_past_samples_used": 0,
|
| 169 |
+
"transfuser_num_bev_cross_attention_layers": 6,
|
| 170 |
+
"transfuser_num_bev_cross_attention_heads": 8,
|
| 171 |
+
"transfuser_token_dim": 256,
|
| 172 |
+
"predict_target_speed": false,
|
| 173 |
+
"predict_spatial_path": false,
|
| 174 |
+
"predict_temporal_spatial_waypoints": true,
|
| 175 |
+
"use_planning_decoder": true,
|
| 176 |
+
"target_speed_classes": [
|
| 177 |
+
0.0,
|
| 178 |
+
4.0,
|
| 179 |
+
8.0,
|
| 180 |
+
10.0,
|
| 181 |
+
13.88888888,
|
| 182 |
+
16.0,
|
| 183 |
+
17.77777777,
|
| 184 |
+
20.0
|
| 185 |
+
],
|
| 186 |
+
"smooth_route": true,
|
| 187 |
+
"num_route_points_smoothing": 20,
|
| 188 |
+
"num_route_points_prediction": 10,
|
| 189 |
+
"num_way_points_prediction": 8,
|
| 190 |
+
"waypoints_spacing": 10,
|
| 191 |
+
"crop_height": 10,
|
| 192 |
+
"carla_crop_height_type": 1,
|
| 193 |
+
"image_width_before_camera_subselection": 1920,
|
| 194 |
+
"final_image_width": 1920,
|
| 195 |
+
"final_image_height": 270,
|
| 196 |
+
"img_vert_anchors": 8,
|
| 197 |
+
"img_horz_anchors": 60,
|
| 198 |
+
"freeze_backbone": false,
|
| 199 |
+
"image_architecture": "resnet34",
|
| 200 |
+
"lidar_architecture": "resnet34",
|
| 201 |
+
"LTF": true,
|
| 202 |
+
"num_fusion_stages": 4,
|
| 203 |
+
"only_ego_status_input": false,
|
| 204 |
+
"block_exp": 4,
|
| 205 |
+
"n_layer": 2,
|
| 206 |
+
"n_head": 4,
|
| 207 |
+
"embd_pdrop": 0.1,
|
| 208 |
+
"resid_pdrop": 0.1,
|
| 209 |
+
"attn_pdrop": 0.1,
|
| 210 |
+
"gpt_linear_layer_init_mean": 0.0,
|
| 211 |
+
"gpt_linear_layer_init_std": 0.02,
|
| 212 |
+
"gpt_layer_norm_init_weight": 1.0,
|
| 213 |
+
"use_semantic": true,
|
| 214 |
+
"num_semantic_classes": 10,
|
| 215 |
+
"perspective_downsample_factor": 1,
|
| 216 |
+
"deconv_channel_num_0": 128,
|
| 217 |
+
"deconv_channel_num_1": 64,
|
| 218 |
+
"deconv_channel_num_2": 32,
|
| 219 |
+
"deconv_scale_factor_0": 4,
|
| 220 |
+
"deconv_scale_factor_1": 8,
|
| 221 |
+
"use_bev_semantic": true,
|
| 222 |
+
"num_bev_semantic_classes": 13,
|
| 223 |
+
"scale_pedestrian_bev_semantic_size": 2.5,
|
| 224 |
+
"pedestrian_bev_min_extent": 0.4,
|
| 225 |
+
"bev_features_chanels": 64,
|
| 226 |
+
"bev_down_sample_factor": 4,
|
| 227 |
+
"bev_upsample_factor": 2,
|
| 228 |
+
"use_carla_data": false,
|
| 229 |
+
"carla_num_samples": 104000,
|
| 230 |
+
"use_navsim_data": true,
|
| 231 |
+
"navsim_data_root": "data/navsim_training_cache/trainval",
|
| 232 |
+
"navsim_num_samples": 104000,
|
| 233 |
+
"schedule_carla_num_samples": false,
|
| 234 |
+
"use_waymo_e2e_data": false,
|
| 235 |
+
"waymo_e2e_num_training_samples": -1,
|
| 236 |
+
"waymo_e2e_training_data_root": "data/waymo_open_dataset_end_to_end_camera_v_1_0_0_training",
|
| 237 |
+
"waymo_e2e_val_data_root": "data/waymo_open_dataset_end_to_end_camera_v_1_0_0_val",
|
| 238 |
+
"waymo_e2e_test_data_root": "data/waymo_open_dataset_end_to_end_camera_v_1_0_0_test",
|
| 239 |
+
"waymo_e2e_subsample_factor": 5,
|
| 240 |
+
"navsim_num_bev_semantic_classes": 7,
|
| 241 |
+
"navsim_num_bb_classes": 1,
|
| 242 |
+
"mixed_data_training": false,
|
| 243 |
+
"carla_leaderboard_mode": false,
|
| 244 |
+
"log_scalars_frequency": 50,
|
| 245 |
+
"log_images_frequency": 2000,
|
| 246 |
+
"slurm_job_id": "1909240",
|
| 247 |
+
"log_wandb": true,
|
| 248 |
+
"gpu_name": "l40s",
|
| 249 |
+
"rank": 0,
|
| 250 |
+
"world_size": 1,
|
| 251 |
+
"local_rank": 0,
|
| 252 |
+
"assigned_cpu_cores": 16,
|
| 253 |
+
"workers_per_cpu_cores": 1
|
| 254 |
+
}
|