fix(nyz): fix structured action bug (#760)

opendilab · Dec 21, 2023 · a2b5ab7 · a2b5ab7
1 parent 9116ba6
commit a2b5ab7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ding/policy/common_utils.py b/ding/policy/common_utils.py
@@ -29,7 +29,8 @@ def default_preprocess_learn(
             the following model forward and loss computation.
     """
     # data preprocess
-    if data[0]['action'].dtype in [np.int64, torch.int64]:
+    elem = data[0]
+    if isinstance(elem['action'], torch.Tensor) and elem['action'].dtype in [np.int64, torch.int64]:
         data = default_collate(data, cat_1dim=True)  # for discrete action
     else:
         data = default_collate(data, cat_1dim=False)  # for continuous action