(params, lrs_dict, tracking)
| 133 | |
| 134 | |
| 135 | def initialize_optimizer(params, lrs_dict, tracking): |
| 136 | lrs = lrs_dict |
| 137 | param_groups = [{'params': [v], 'name': k, 'lr': lrs[k]} for k, v in params.items()] |
| 138 | if tracking: |
| 139 | return torch.optim.Adam(param_groups) |
| 140 | else: |
| 141 | return torch.optim.Adam(param_groups, lr=0.0, eps=1e-15) |
| 142 | |
| 143 | def initialize_first_timestep(dataset, num_frames, scene_radius_depth_ratio, mean_sq_dist_method, seg_net): |
| 144 | # Get RGB-D Data & semantic & Camera Parameters |
no outgoing calls
no test coverage detected