Skip to content

Commit

Permalink
reverting to cpu image caching
Browse files Browse the repository at this point in the history
  • Loading branch information
liorf committed Jan 4, 2024
1 parent 566d670 commit 5caeadc
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion nerfstudio/data/datamanagers/full_images_datamanager.py
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,7 @@ class FullImageDatamanagerConfig(DataManagerConfig):
new images. If -1, never pick new images."""
eval_image_indices: Optional[Tuple[int, ...]] = (0,)
"""Specifies the image indices to use during eval; if None, uses all."""
cache_images: Literal["no-cache", "cpu", "gpu"] = "gpu"
cache_images: Literal["no-cache", "cpu", "gpu"] = "cpu"
"""Whether to cache images in memory. If "numpy", caches as numpy arrays, if "torch", caches as torch tensors."""


Expand Down

0 comments on commit 5caeadc

Please sign in to comment.