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))
|