File size: 5,711 Bytes
0b72465 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 |
{
"accumulate_grad_batches": 1,
"add_external_cam": false,
"apply_geometry_noise": 0,
"apply_random_transforms": 0,
"base_log_dir": {
"_string": "/eval/train_logs",
"_type": "python_object (type = PosixPath)",
"_value": "gASVNQAAAAAAAACMB3BhdGhsaWKUjAlQb3NpeFBhdGiUk5SMAS+UjARldmFslIwKdHJhaW5fbG9nc5SHlFKULg=="
},
"batch_size": 16,
"batch_size_val": 16,
"checkpoint": null,
"data_type": {
"_string": "DataType.RGBD_AND_MESH",
"_type": "python_object (type = DataType)",
"_value": "gASVRAAAAAAAAACMH21pbmRtYXAuZGF0YV9sb2FkaW5nLmRhdGFfdHlwZXOUjAhEYXRhVHlwZZSTlIwNcmdiZF9hbmRfbWVzaJSFlFKULg=="
},
"dataset": "/datasets/mindmap_cubes_rgbdmesh_radioB_1cam/",
"demos_datagen": "0",
"demos_train": "0-7",
"demos_valset": "8-9",
"diffusion_dropout": 0.0,
"diffusion_timesteps": 100,
"disable_visualizer_wait_on_key": false,
"embedding_dim": 120,
"encode_openness": 1,
"encoder_dropout": 0.0,
"eval_only": false,
"exp_name": "mindmap Training",
"extra_keyposes_around_grasp_events": null,
"feature_image_size": {
"_type": "tuple",
"_value": [
32,
32
]
},
"feature_type": {
"_string": "FeatureExtractorType.RADIO_V25_B",
"_type": "python_object (type = FeatureExtractorType)",
"_value": "gASVWgAAAAAAAACMK21pbmRtYXAuaW1hZ2VfcHJvY2Vzc2luZy5mZWF0dXJlX2V4dHJhY3Rpb26UjBRGZWF0dXJlRXh0cmFjdG9yVHlwZZSTlIwLcmFkaW9fdjI1X2KUhZRSlC4="
},
"fpn_checkpoint": null,
"fps_subsampling_factor": 5,
"gripper_encoding_mode": "binary",
"gripper_loss": 1.0,
"ignore_model_args_json": false,
"image_size": {
"_type": "tuple",
"_value": [
512,
512
]
},
"include_dynamic": false,
"include_failed_demos": false,
"initial_learning_rate": 0.0001,
"instructions": null,
"keypose_detection_mode": null,
"lang_enhanced": 0,
"learning_rate_convergence_percentage": 0.75,
"learning_rate_end_factor": 0.5,
"max_episode_length": 5,
"max_episodes_per_task": 100,
"num_batches_per_test_eval": 1,
"num_batches_per_train_eval": 1,
"num_history": 3,
"num_vertices_to_sample": 2048,
"num_vis_ins_attn_layers": 2,
"num_workers": 0,
"num_workers_for_test_dataset": null,
"only_sample_keyposes": false,
"pos_loss": 30.0,
"pos_noise_stddev_m": 0.01,
"prediction_horizon": 1,
"predictor_dropout": 0.0,
"print_progress_freq": 100,
"print_timers_freq": 1000,
"projective_appearance_integrator_measurement_weight": null,
"quaternion_format": "wxyz",
"random_rpy_range_deg": {
"_type": "tuple",
"_value": [
[
0.0,
0.0,
-90.0
],
[
0.0,
0.0,
90.0
]
]
},
"random_translation_range_m": {
"_type": "tuple",
"_value": [
[
-0.1,
-0.1,
0.0
],
[
0.1,
0.1,
0.0
]
]
},
"relative_action": 0,
"reproducibility": {
"command_line": "python run_training.py --task cube_stacking --data_type rgbd_and_mesh --feature_type radio_v25_b --demos_train 0-7 --demos_valset 8-9 --dataset /datasets/mindmap_cubes_rgbdmesh_radioB_1cam/",
"git_has_uncommitted_changes": true,
"git_root": "/workspaces/mindmap",
"git_url": "https://gitlab-master.nvidia.com/12051/nvblox/mindmap/tree/12fa2bdc1462db417618cd17a0a08a4a57a883cb",
"time": "Tue Sep 9 18:59:03 2025"
},
"rgbd_min_depth_threshold": 0.0,
"rot_loss": 10.0,
"rot_noise_stddev_deg": 0.01,
"rotation_parametrization": "6D_from_query",
"sampling_weighting_type": "uniform",
"save_checkpoint": true,
"save_serialized_nvblox_map_to_disk": false,
"seed": 0,
"skip_train_val": false,
"task": {
"_string": "Tasks.CUBE_STACKING",
"_type": "python_object (type = Tasks)",
"_value": "gASVNQAAAAAAAACME21pbmRtYXAudGFza3MudGFza3OUjAVUYXNrc5STlIwNY3ViZV9zdGFja2luZ5SFlFKULg=="
},
"train_iters": 24000,
"use_fps": 1,
"use_instruction": 0,
"use_keyposes": 1,
"use_shared_feature_encoder": 0,
"val_freq": 12,
"validate_demos_with_gt_poses": 1,
"variations": {
"_type": "tuple",
"_value": [
0
]
},
"vertex_sampling_method": {
"_string": "VertexSamplingMethod.RANDOM_WITHOUT_REPLACEMENT",
"_type": "python_object (type = VertexSamplingMethod)",
"_value": "gASVYgAAAAAAAACMJG1pbmRtYXAuZGF0YV9sb2FkaW5nLnZlcnRleF9zYW1wbGluZ5SMFFZlcnRleFNhbXBsaW5nTWV0aG9klJOUjBpyYW5kb21fd2l0aG91dF9yZXBsYWNlbWVudJSFlFKULg=="
},
"visualize": false,
"visualize_aabb": false,
"visualize_attention_weights": false,
"visualize_backprojected_features": false,
"visualize_encoded_features": true,
"visualizer_background_rgb": {
"_type": "tuple",
"_value": [
0.0,
0.0,
0.0
]
},
"visualizer_min_attention_weight": 0.0,
"visualizer_min_tsdf_weight": 0.0,
"visualizer_point_size": 3,
"visualizer_pointcloud_max_distance": null,
"visualizer_pointclouds_ply_output_dir": null,
"visualizer_record_camera_output_path": null,
"visualizer_voxel_size_m": 0.01,
"viz_freq": 200,
"voxel_size_m": null,
"wandb_entity": "nv-welcome",
"wandb_mode": "online",
"wandb_name": null
} |