update

agent-dreamer_v3.py

@@ -79,7 +79,7 @@ def main(cfg_path: str, checkpoint_path: str, test=False):
 
     while True:
         # Convert numpy observations into torch observations and normalize image observations
-        torch_obs = prepare_obs(fabric, o, cnn_keys)
+        torch_obs = prepare_obs(fabric, o, cnn_keys=cnn_keys)
 
         # Select actions, the agent returns a one-hot categorical or
         # more one-hot categorical distributions for muli-discrete actions space

agent-ppo.py

@@ -69,7 +69,7 @@ def main(cfg_path: str, checkpoint_path: str, test=False):
 
     while True:
         # Convert numpy observations into torch observations and normalize image observations
-        torch_obs = prepare_obs(fabric, o, cnn_keys)
+        torch_obs = prepare_obs(fabric, o, cnn_keys=cnn_keys)
 
         actions = agent.get_actions(torch_obs, greedy=True)
         actions = torch.cat([act.argmax(dim=-1) for act in actions], dim=-1)