[RLlib] Dreamer (#10172)

2026-06-28 16:30:41 +08:00 · 2020-08-26 04:24:05 -07:00
parent 9ca159aa0b
commit 4e9888ce2f
13 changed files with 1362 additions and 69 deletions
@@ -227,6 +227,13 @@ class TBXLogger(Logger):
                  and len(value) > 0) or (type(value) == np.ndarray
                                          and value.size > 0):
                valid_result[full_attr] = value
+
+                # Must be video
+                if type(value) == np.ndarray and value.ndim == 5:
+                    self._file_writer.add_video(
+                        full_attr, value, global_step=step, fps=20)
+                    continue
+
                try:
                    self._file_writer.add_histogram(
                        full_attr, value, global_step=step)