(self,
encoder: Dict[str, Any],
neck: Dict[str, Any],
points_head: Dict[str, Any] = None,
mask_head: Dict[str, Any] = None,
normal_head: Dict[str, Any] = None,
scale_head: Dict[str, Any] = None,
remap_output: Literal['linear', 'sinh', 'exp', 'sinh_exp'] = 'linear',
num_tokens_range: List[int] = [1200, 3600],
**deprecated_kwargs
)
| 27 | scale_head: MLP |
| 28 | |
| 29 | def __init__(self, |
| 30 | encoder: Dict[str, Any], |
| 31 | neck: Dict[str, Any], |
| 32 | points_head: Dict[str, Any] = None, |
| 33 | mask_head: Dict[str, Any] = None, |
| 34 | normal_head: Dict[str, Any] = None, |
| 35 | scale_head: Dict[str, Any] = None, |
| 36 | remap_output: Literal['linear', 'sinh', 'exp', 'sinh_exp'] = 'linear', |
| 37 | num_tokens_range: List[int] = [1200, 3600], |
| 38 | **deprecated_kwargs |
| 39 | ): |
| 40 | super(MoGeModel, self).__init__() |
| 41 | if deprecated_kwargs: |
| 42 | warnings.warn(f"The following deprecated/invalid arguments are ignored: {deprecated_kwargs}") |
| 43 | |
| 44 | self.remap_output = remap_output |
| 45 | self.num_tokens_range = num_tokens_range |
| 46 | |
| 47 | self.encoder = DINOv2Encoder(**encoder) |
| 48 | self.neck = ConvStack(**neck) |
| 49 | if points_head is not None: |
| 50 | self.points_head = ConvStack(**points_head) |
| 51 | if mask_head is not None: |
| 52 | self.mask_head = ConvStack(**mask_head) |
| 53 | if normal_head is not None: |
| 54 | self.normal_head = ConvStack(**normal_head) |
| 55 | if scale_head is not None: |
| 56 | self.scale_head = MLP(**scale_head) |
| 57 | |
| 58 | @property |
| 59 | def device(self) -> torch.device: |
nothing calls this directly
no test coverage detected