xizaoqu
commited on
Commit
·
70eef9b
1
Parent(s):
e05d15b
update README
Browse files
app.py
CHANGED
@@ -223,6 +223,8 @@ def run(cfg: DictConfig):
|
|
223 |
memory_curr_frame,
|
224 |
device
|
225 |
)
|
|
|
|
|
226 |
|
227 |
memory_frames.append(new_frame)
|
228 |
|
@@ -244,8 +246,8 @@ def run(cfg: DictConfig):
|
|
244 |
global input_history
|
245 |
global memory_frames
|
246 |
|
247 |
-
device = next(runner.algo.parameters()).device
|
248 |
-
print(device)
|
249 |
runner.algo.reset()
|
250 |
memory_frames = []
|
251 |
memory_frames.append(load_image_as_tensor(DEFAULT_IMAGE))
|
|
|
223 |
memory_curr_frame,
|
224 |
device
|
225 |
)
|
226 |
+
|
227 |
+
print("algo frame:", len(runner.algo.frames))
|
228 |
|
229 |
memory_frames.append(new_frame)
|
230 |
|
|
|
246 |
global input_history
|
247 |
global memory_frames
|
248 |
|
249 |
+
# device = next(runner.algo.parameters()).device
|
250 |
+
# print(device)
|
251 |
runner.algo.reset()
|
252 |
memory_frames = []
|
253 |
memory_frames.append(load_image_as_tensor(DEFAULT_IMAGE))
|