From 48585c525ee7e5d1e029fa867de5657631e0cc43 Mon Sep 17 00:00:00 2001 From: Hoanh Le Date: Fri, 30 Aug 2024 17:18:18 +0300 Subject: [PATCH] change grid resolution type --- nerfstudio/models/instant_ngp.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nerfstudio/models/instant_ngp.py b/nerfstudio/models/instant_ngp.py index 2503e7e228..48cad339d0 100644 --- a/nerfstudio/models/instant_ngp.py +++ b/nerfstudio/models/instant_ngp.py @@ -19,7 +19,7 @@ from __future__ import annotations from dataclasses import dataclass, field -from typing import Dict, List, Literal, Optional, Tuple, Type +from typing import Dict, List, Literal, Optional, Tuple, Type, Union import nerfacc import torch @@ -49,7 +49,7 @@ class InstantNGPModelConfig(ModelConfig): """Whether to create a scene collider to filter rays.""" collider_params: Optional[Dict[str, float]] = None """Instant NGP doesn't use a collider.""" - grid_resolution: int = 128 + grid_resolution: Union[int, List[int]] = 128 """Resolution of the grid used for the field.""" grid_levels: int = 4 """Levels of the grid used for the field."""