| 32 | |
| 33 | @dataclass |
| 34 | class ModelExports: |
| 35 | extrinsics: Float[Tensor, "batch frame 4 4"] |
| 36 | intrinsics: Float[Tensor, "batch frame 3 3"] |
| 37 | colors: Float[Tensor, "batch frame 3 height width"] |
| 38 | depths: Float[Tensor, "batch frame height width"] |
| 39 | |
| 40 | |
| 41 | class Model(nn.Module): |