mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-11 18:44:21 +08:00
fixed memory leakage caused by writer (#250)
* add print * updated benchmark for more fine-grain options * fixed increasing gpu memory leakage * merge didnt save * isort not working locally? * checking isort order
This commit is contained in:
1 parent
fce10e6bc8
commit
113b6d3faf
1 file changed
+1
-1
@@ -68,7 +68,7 @@ def put_image(name, image: TensorType["H", "W", "C"], step: int):
|
||||
if isinstance(name, EventName):
|
||||
name = name.value
|
||||
|
||||
EVENT_STORAGE.append({"name": name, "write_type": EventType.IMAGE, "event": image, "step": step})
|
||||
EVENT_STORAGE.append({"name": name, "write_type": EventType.IMAGE, "event": image.detach().cpu(), "step": step})
|
||||
|
||||
|
||||
@check_main_thread
|
||||
|
||||
Reference in new issue
Block a user