xizaoqu
commited on
Commit
·
1cccab7
1
Parent(s):
76b382c
update README
Browse files
app.py
CHANGED
@@ -176,8 +176,8 @@ def run(cfg: DictConfig):
|
|
176 |
algo = run_local(cfg)
|
177 |
algo.to(device)
|
178 |
|
179 |
-
|
180 |
-
print("algo:",
|
181 |
|
182 |
actions = torch.zeros((1, 25))
|
183 |
poses = torch.zeros((1, 5))
|
@@ -186,8 +186,8 @@ def run(cfg: DictConfig):
|
|
186 |
|
187 |
runner = InteractiveRunner(algo)
|
188 |
|
189 |
-
|
190 |
-
print("runner.algo:",
|
191 |
|
192 |
# @spaces.GPU()
|
193 |
# def run_interactive(first_frame, action, first_pose, curr_frame, device):
|
|
|
176 |
algo = run_local(cfg)
|
177 |
algo.to(device)
|
178 |
|
179 |
+
algodevice = next(algo.parameters()).device
|
180 |
+
print("algo:", algodevice)
|
181 |
|
182 |
actions = torch.zeros((1, 25))
|
183 |
poses = torch.zeros((1, 5))
|
|
|
186 |
|
187 |
runner = InteractiveRunner(algo)
|
188 |
|
189 |
+
algodevice = next(runner.algo.parameters()).device
|
190 |
+
print("runner.algo:", algodevice)
|
191 |
|
192 |
# @spaces.GPU()
|
193 |
# def run_interactive(first_frame, action, first_pose, curr_frame, device):
|