Upload folder using huggingface_hub
Browse files
tfv6_regnety032/config.json
ADDED
|
@@ -0,0 +1,237 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"visualize_dataset": false,
|
| 3 |
+
"visualize_failed_scenarios": false,
|
| 4 |
+
"load_bev_3rd_person_images": false,
|
| 5 |
+
"seed": 0,
|
| 6 |
+
"wandb_id": "733_scaled_regnety_010_postrain32_0_251025_182327",
|
| 7 |
+
"wandb_resume": "allow",
|
| 8 |
+
"description": "733_scaled_regnety_010_postrain32_0_251025_182327",
|
| 9 |
+
"visualize_training": true,
|
| 10 |
+
"id": "733_scaled_regnety_010_postrain32_0_251025_182327",
|
| 11 |
+
"load_file": "/home/nguyen/master_thesis/outputs/training/733_scaled_regnety/000_pretrain1_0/251018_092144/model_0030.pth",
|
| 12 |
+
"continue_epoch": false,
|
| 13 |
+
"delete_old_checkpoints": true,
|
| 14 |
+
"use_ssd_cache": true,
|
| 15 |
+
"use_hdd_cache": true,
|
| 16 |
+
"force_rebuild_data_cache": false,
|
| 17 |
+
"cache_characteristics": [
|
| 18 |
+
"1152",
|
| 19 |
+
"384",
|
| 20 |
+
"-32",
|
| 21 |
+
"64",
|
| 22 |
+
"-40",
|
| 23 |
+
"40",
|
| 24 |
+
"True",
|
| 25 |
+
"True",
|
| 26 |
+
"True",
|
| 27 |
+
"True",
|
| 28 |
+
"False",
|
| 29 |
+
"10",
|
| 30 |
+
"8",
|
| 31 |
+
"5"
|
| 32 |
+
],
|
| 33 |
+
"ssd_cache_path": "/scratch/1866984",
|
| 34 |
+
"carla_root": "data/carla_leaderboad2_v8/results/data/sensor_data/",
|
| 35 |
+
"carla_data": "data/carla_leaderboad2_v8/results/data/sensor_data/data",
|
| 36 |
+
"logdir": "/home/nguyen/master_thesis/outputs/training/733_scaled_regnety/010_postrain32_0/251025_182327",
|
| 37 |
+
"training_png_compression_level": 6,
|
| 38 |
+
"vehicle_min_num_lidar_points": 1,
|
| 39 |
+
"vehicle_min_num_visible_pixels": 1,
|
| 40 |
+
"pedestrian_min_num_lidar_points": 5,
|
| 41 |
+
"pedestrian_min_num_visible_pixels": 15,
|
| 42 |
+
"parking_vehicle_min_num_lidar_points": 3,
|
| 43 |
+
"parking_vehicle_min_num_visible_pixels": 5,
|
| 44 |
+
"grad_scaler_init_scale": 1024,
|
| 45 |
+
"grad_scaler_growth_factor": 2,
|
| 46 |
+
"grad_scaler_backoff_factor": 0.5,
|
| 47 |
+
"grad_scaler_growth_interval": 256,
|
| 48 |
+
"grad_scaler_max_grad_scale": 65536,
|
| 49 |
+
"sync_batchnorm": false,
|
| 50 |
+
"epochs": 31,
|
| 51 |
+
"epochs_exit": 31,
|
| 52 |
+
"batch_size": 64,
|
| 53 |
+
"use_mixed_precision_training": true,
|
| 54 |
+
"need_grad_scaler": false,
|
| 55 |
+
"use_zero_redundancy": false,
|
| 56 |
+
"save_model_checkpoint": true,
|
| 57 |
+
"is_pretraining": false,
|
| 58 |
+
"prefetch_factor": 8,
|
| 59 |
+
"compile": true,
|
| 60 |
+
"channel_last": true,
|
| 61 |
+
"lr": 0.0003,
|
| 62 |
+
"warmup_epochs": 1,
|
| 63 |
+
"skip_first": 10,
|
| 64 |
+
"radar_detection": true,
|
| 65 |
+
"use_radar_detection": true,
|
| 66 |
+
"num_radar_points_per_sensor": 75,
|
| 67 |
+
"num_radar_queries": 20,
|
| 68 |
+
"radar_hidden_dim_tokenizer": 1024,
|
| 69 |
+
"radar_token_dim": 256,
|
| 70 |
+
"radar_tf_dim_ff": 1024,
|
| 71 |
+
"radar_dropout": 0.1,
|
| 72 |
+
"radar_num_heads": 8,
|
| 73 |
+
"radar_num_layers": 4,
|
| 74 |
+
"radar_hidden_dim_decoder": 1024,
|
| 75 |
+
"radar_classification_loss_weight": 1.0,
|
| 76 |
+
"radar_forecast_loss_weight": 0.1,
|
| 77 |
+
"radar_regression_loss_weight": 5.0,
|
| 78 |
+
"radar_sine_features": 128,
|
| 79 |
+
"num_radar_sensors": 4,
|
| 80 |
+
"forecast_radar_detections": false,
|
| 81 |
+
"num_radar_forecast_steps": 8,
|
| 82 |
+
"radar_waypoints_spacing": 5,
|
| 83 |
+
"max_distance_future_waypoint": 10.0,
|
| 84 |
+
"force_rebuild_bucket_cache": false,
|
| 85 |
+
"resample_posttrain_buckets": false,
|
| 86 |
+
"randomize_route_order": false,
|
| 87 |
+
"subsample_data": false,
|
| 88 |
+
"bucket_collection": "full_posttrain",
|
| 89 |
+
"bucket_collection_path": "data/carla_leaderboad2_v8/results/data/sensor_data/buckets",
|
| 90 |
+
"augment": true,
|
| 91 |
+
"augment_percentage": 0.5,
|
| 92 |
+
"training_extreme_augment_percentage": 0.0,
|
| 93 |
+
"use_color_aug": true,
|
| 94 |
+
"color_aug_prob": 0.2,
|
| 95 |
+
"weight_decay": 0.01,
|
| 96 |
+
"use_grad_clip": false,
|
| 97 |
+
"grad_quantile": 0.975,
|
| 98 |
+
"grad_history_length": 5000,
|
| 99 |
+
"use_optimizer_groups": false,
|
| 100 |
+
"use_depth": true,
|
| 101 |
+
"training_used_lidar_steps": 10,
|
| 102 |
+
"min_z": -4,
|
| 103 |
+
"max_z": 4,
|
| 104 |
+
"pixels_per_meter": 4.0,
|
| 105 |
+
"hist_max_per_pixel": 5,
|
| 106 |
+
"lidar_resolution_width": 384,
|
| 107 |
+
"lidar_resolution_height": 320,
|
| 108 |
+
"lidar_width": 96,
|
| 109 |
+
"lidar_height": 80,
|
| 110 |
+
"lidar_vert_anchors": 10,
|
| 111 |
+
"lidar_horz_anchors": 12,
|
| 112 |
+
"detect_boxes": true,
|
| 113 |
+
"debug_boxes_visualization": true,
|
| 114 |
+
"center_net_global_avg_factor": false,
|
| 115 |
+
"data_bb_static_types_white_list": [
|
| 116 |
+
"static.prop.constructioncone",
|
| 117 |
+
"static.prop.trafficwarning"
|
| 118 |
+
],
|
| 119 |
+
"bb_confidence_threshold": 0.3,
|
| 120 |
+
"max_num_bbs": 90,
|
| 121 |
+
"num_dir_bins": 12,
|
| 122 |
+
"top_k_center_keypoints": 100,
|
| 123 |
+
"center_net_max_pooling_kernel": 3,
|
| 124 |
+
"bb_input_channel": 64,
|
| 125 |
+
"angle_bins": [
|
| 126 |
+
-0.375,
|
| 127 |
+
-0.125,
|
| 128 |
+
0.125,
|
| 129 |
+
0.375
|
| 130 |
+
],
|
| 131 |
+
"angles": [
|
| 132 |
+
-0.5,
|
| 133 |
+
-0.25,
|
| 134 |
+
0.0,
|
| 135 |
+
0.25,
|
| 136 |
+
0.5
|
| 137 |
+
],
|
| 138 |
+
"car_open_door_extra_width": 1.2,
|
| 139 |
+
"traffic_warning_bb_size": [
|
| 140 |
+
1.186714768409729,
|
| 141 |
+
1.4352929592132568
|
| 142 |
+
],
|
| 143 |
+
"construction_cone_bb_size": [
|
| 144 |
+
0.1720348298549652,
|
| 145 |
+
0.1720348298549652
|
| 146 |
+
],
|
| 147 |
+
"num_bb_classes": 8,
|
| 148 |
+
"use_discrete_command": true,
|
| 149 |
+
"use_noisy_tp": false,
|
| 150 |
+
"use_velocity": true,
|
| 151 |
+
"use_previous_tp": true,
|
| 152 |
+
"use_next_tp": true,
|
| 153 |
+
"max_speed": 25.0,
|
| 154 |
+
"target_points_normalization_constants": [
|
| 155 |
+
[
|
| 156 |
+
200.0,
|
| 157 |
+
50.0
|
| 158 |
+
]
|
| 159 |
+
],
|
| 160 |
+
"tp_pop_distance": 3.25,
|
| 161 |
+
"transfuser_num_bev_cross_attention_layers": 6,
|
| 162 |
+
"transfuser_num_bev_cross_attention_heads": 8,
|
| 163 |
+
"transfuser_token_dim": 256,
|
| 164 |
+
"use_planning_decoder": true,
|
| 165 |
+
"target_speeds": [
|
| 166 |
+
0.0,
|
| 167 |
+
4.0,
|
| 168 |
+
8.0,
|
| 169 |
+
10.0,
|
| 170 |
+
13.88888888,
|
| 171 |
+
16.0,
|
| 172 |
+
17.77777777,
|
| 173 |
+
20.0
|
| 174 |
+
],
|
| 175 |
+
"smooth_route": true,
|
| 176 |
+
"num_route_points_smoothing": 20,
|
| 177 |
+
"num_route_points_prediction": 10,
|
| 178 |
+
"num_way_points_prediction": 8,
|
| 179 |
+
"waypoints_spacing": 5,
|
| 180 |
+
"image_resolution": 1.0,
|
| 181 |
+
"final_image_width": 1152,
|
| 182 |
+
"final_image_height": 384,
|
| 183 |
+
"img_vert_anchors": 12,
|
| 184 |
+
"img_horz_anchors": 36,
|
| 185 |
+
"freeze_backbone": false,
|
| 186 |
+
"image_architecture": "regnety_032",
|
| 187 |
+
"lidar_architecture": "regnety_032",
|
| 188 |
+
"block_exp": 4,
|
| 189 |
+
"n_layer": 2,
|
| 190 |
+
"n_head": 4,
|
| 191 |
+
"embd_pdrop": 0.1,
|
| 192 |
+
"resid_pdrop": 0.1,
|
| 193 |
+
"attn_pdrop": 0.1,
|
| 194 |
+
"gpt_linear_layer_init_mean": 0.0,
|
| 195 |
+
"gpt_linear_layer_init_std": 0.02,
|
| 196 |
+
"gpt_layer_norm_init_weight": 1.0,
|
| 197 |
+
"use_semantic": true,
|
| 198 |
+
"num_semantic_classes": 10,
|
| 199 |
+
"perspective_downsample_factor": 1,
|
| 200 |
+
"deconv_channel_num_0": 128,
|
| 201 |
+
"deconv_channel_num_1": 64,
|
| 202 |
+
"deconv_channel_num_2": 32,
|
| 203 |
+
"deconv_scale_factor_0": 4,
|
| 204 |
+
"deconv_scale_factor_1": 8,
|
| 205 |
+
"use_bev_semantic": true,
|
| 206 |
+
"num_bev_semantic_classes": 13,
|
| 207 |
+
"scale_pedestrian_bev_semantic_size": 2.5,
|
| 208 |
+
"pedestrian_bev_min_extent": 0.4,
|
| 209 |
+
"bev_features_chanels": 64,
|
| 210 |
+
"bev_down_sample_factor": 4,
|
| 211 |
+
"bev_upsample_factor": 2,
|
| 212 |
+
"detailed_loss_weights": {
|
| 213 |
+
"loss_tf": 0.08333326388894677,
|
| 214 |
+
"loss_tfpp_speed": 0.08333326388894677,
|
| 215 |
+
"loss_tfpp": 0.08333326388894677,
|
| 216 |
+
"loss_semantic": 0.08333326388894677,
|
| 217 |
+
"loss_depth": 8.333326388894678e-07,
|
| 218 |
+
"loss_bev_semantic": 0.08333326388894677,
|
| 219 |
+
"loss_center_net_heatmap": 0.08333326388894677,
|
| 220 |
+
"loss_center_net_wh": 0.08333326388894677,
|
| 221 |
+
"loss_center_net_offset": 0.08333326388894677,
|
| 222 |
+
"loss_center_net_yaw_class": 0.08333326388894677,
|
| 223 |
+
"loss_center_net_yaw_res": 0.08333326388894677,
|
| 224 |
+
"loss_center_net_velocity": 0.08333326388894677,
|
| 225 |
+
"radar_loss": 0.08333326388894677
|
| 226 |
+
},
|
| 227 |
+
"log_scalars_frequency": 50,
|
| 228 |
+
"log_images_frequency": 500,
|
| 229 |
+
"slurm_job_id": "1866984",
|
| 230 |
+
"log_wandb": true,
|
| 231 |
+
"gpu_name": "l40s",
|
| 232 |
+
"rank": 0,
|
| 233 |
+
"world_size": 4,
|
| 234 |
+
"local_rank": 0,
|
| 235 |
+
"assigned_cpu_cores": 64,
|
| 236 |
+
"workers_per_cpu_cores": 1
|
| 237 |
+
}
|
tfv6_regnety032/model_0030_0.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:94468196d3cd55b99ce22df9aeb88f478f83b0dd7af78fcb8e737bad30d1c76c
|
| 3 |
+
size 506005098
|
tfv6_regnety032/model_0030_1.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3fcbf59ec6ff6c4f1ffb86a3db838fdf734ad781e09747ad0732f97daaa37819
|
| 3 |
+
size 506005098
|
tfv6_regnety032/model_0030_2.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5bcc8e68432c8afe783ab1264a51766cc284af818c37020d08be76a7dc163cf3
|
| 3 |
+
size 506005098
|