mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-14 20:03:49 +08:00
1 parent
f37351ee7f
commit
177c3dd6f1
1 file changed
-2
@@ -29,8 +29,6 @@ def get_dict_to_torch(stuff: Any, device: Union[torch.device, str] = "cpu", excl
|
||||
|
||||
Args:
|
||||
stuff: things to convert to torch
|
||||
device: machine to put the "stuff" on
|
||||
exclude: list of keys to skip over transferring to device
|
||||
"""
|
||||
if isinstance(stuff, dict):
|
||||
for k, v in stuff.items():
|
||||
|
||||
Reference in new issue
Block a user