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