DHDRL commited on
Commit
fff45d3
·
verified ·
1 Parent(s): ba8fcef

Update train_curriculum.py

Browse files
Files changed (1) hide show
  1. train_curriculum.py +4 -1
train_curriculum.py CHANGED
@@ -352,7 +352,10 @@ class DynaCallback(BaseCallback):
352
  return True # safe: missing locals, skip silently
353
 
354
  if hasattr(obs_now, "numpy"):
355
- obs_now_np = {k: v.cpu().numpy() for k, v in obs_now.items()} if hasattr(obs_now, "items") else {"_raw": obs_now.cpu().numpy()}
 
 
 
356
  else:
357
  obs_now_np = obs_now
358
 
 
352
  return True # safe: missing locals, skip silently
353
 
354
  if hasattr(obs_now, "numpy"):
355
+ if hasattr(obs_now, "items"):
356
+ obs_now_np = {k: v.cpu().numpy() for k, v in obs_now.items()}
357
+ else:
358
+ obs_now_np = {"_raw": obs_now.cpu().numpy()}
359
  else:
360
  obs_now_np = obs_now
361