| from transformers import PretrainedConfig | |
| from typing import List | |
| class MonoSceneConfig(PretrainedConfig): | |
| def __init__( | |
| self, | |
| dataset="kitti", | |
| n_classes=20, | |
| feature=64, | |
| project_scale=2, | |
| full_scene_size=(256, 256, 32), | |
| **kwargs, | |
| ): | |
| self.dataset = dataset | |
| self.n_classes = n_classes | |
| self.feature = feature | |
| self.project_scale = project_scale | |
| self.full_scene_size = full_scene_size | |
| super().__init__(**kwargs) | |