From 113b6d3faf5d777498eba2af4e205b0fc1b304cf Mon Sep 17 00:00:00 2001
From: evonneng <evonneng@users.noreply.github.com>
Date: Thu, 4 Aug 2022 19:22:29 -0700
Subject: [PATCH] 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
---
 nerfactory/utils/writer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nerfactory/utils/writer.py b/nerfactory/utils/writer.py
index acff14dd4..c1a736032 100644
--- a/nerfactory/utils/writer.py
+++ b/nerfactory/utils/writer.py
@@ -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