| 153 | |
| 154 | |
| 155 | def set_optim_info( |
| 156 | config_dict: T.Dict[str, T.Any], |
| 157 | random_drop_rgb_rate: float = 0.5, |
| 158 | random_drop_sample_feature_rate: float = 0.5, |
| 159 | optim_method: str = 'adam', |
| 160 | loss_weight_t: float = 10., |
| 161 | loss_weight_t_l1: float = 0., |
| 162 | loss_weight_normal: float = 1., |
| 163 | loss_weight_normal_l1: float = 0., |
| 164 | loss_weight_plane_normal: float = 0., |
| 165 | loss_weight_plane_normal_l1: float = 0., |
| 166 | loss_weight_hit: float = 1.0, |
| 167 | loss_weight_rgb: float = 1.0, |
| 168 | loss_weight_rgb_normal: float = 0, |
| 169 | loss_weight_rgb_normal_dot: float = 0, |
| 170 | loss_weight_rgb_normal_dot_l1: float = 0, |
| 171 | pcd_noise_std: float = 0, |
| 172 | ) -> T.Dict[str, T.Any]: |
| 173 | config_dict['optim_info']['random_drop_rgb_rate'] = random_drop_rgb_rate |
| 174 | config_dict['optim_info']['random_drop_sample_feature_rate'] = random_drop_sample_feature_rate |
| 175 | config_dict['optim_info']['optim_method'] = optim_method |
| 176 | config_dict['optim_info']['loss_weight_t'] = loss_weight_t |
| 177 | config_dict['optim_info']['loss_weight_t_l1'] = loss_weight_t_l1 |
| 178 | config_dict['optim_info']['loss_weight_normal'] = loss_weight_normal |
| 179 | config_dict['optim_info']['loss_weight_normal_l1'] = loss_weight_normal_l1 |
| 180 | config_dict['optim_info']['loss_weight_plane_normal'] = loss_weight_plane_normal |
| 181 | config_dict['optim_info']['loss_weight_plane_normal_l1'] = loss_weight_plane_normal_l1 |
| 182 | config_dict['optim_info']['loss_weight_hit'] = loss_weight_hit |
| 183 | config_dict['optim_info']['loss_weight_rgb'] = loss_weight_rgb |
| 184 | config_dict['optim_info']['loss_weight_rgb_normal'] = loss_weight_rgb_normal |
| 185 | config_dict['optim_info']['loss_weight_rgb_normal_dot'] = loss_weight_rgb_normal_dot |
| 186 | config_dict['optim_info']['loss_weight_rgb_normal_dot_l1'] = loss_weight_rgb_normal_dot_l1 |
| 187 | config_dict['optim_info']['pcd_noise_std'] = pcd_noise_std |
| 188 | |
| 189 | return config_dict |
| 190 | |
| 191 | |
| 192 | def set_model_info( |