ln2697 commited on
Commit
65d15f1
·
verified ·
1 Parent(s): 9938c62

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