tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 02:24:38 +08:00
Code

fix device typing issue

Browse Source
This commit is contained in:
akristoffersen 2025-02-09 02:29:53 +00:00
parent 7534956d69
commit 9849f64083
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nerfstudio/data/utils/nerfstudio_collate.py
View File

@ -98,7 +98,7 @@ def nerfstudio_collate(batch: Any, extra_mappings: Union[Dict[type, Callable], N
# If we're in a background process, concatenate directly into a
# shared memory tensor to avoid an extra copy
numel = sum(x.numel() for x in batch)
storage = elem.untyped_storage()._new_shared(numel, device=elem.device)
storage = elem.untyped_storage()._new_shared(numel, device=str(elem.device))
out = elem.new(storage).resize_(len(batch), *list(elem.size()))
return torch.stack(batch, 0, out=out)
elif elem_type.__module__ == "numpy" and elem_type.__name__ not in ("str_", "string_"):