xizaoqu commited on
Commit
1cccab7
·
1 Parent(s): 76b382c

update README

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -176,8 +176,8 @@ def run(cfg: DictConfig):
176
  algo = run_local(cfg)
177
  algo.to(device)
178
 
179
- device = next(algo.parameters()).device
180
- print("algo:", device)
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
- device = next(runner.algo.parameters()).device
190
- print("runner.algo:", device)
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):