(
self,
n_channels: int = 3,
**kwargs,
)
| 197 | |
| 198 | class MLPTextureFieldModel(MLPModel): |
| 199 | def __init__( |
| 200 | self, |
| 201 | n_channels: int = 3, |
| 202 | **kwargs, |
| 203 | ): |
| 204 | super().__init__(n_output=n_channels, output_activation="sigmoid", **kwargs) |
| 205 | |
| 206 | def forward( |
| 207 | self, |