[fix] time list

2026-04-12 05:46:22 +02:00 · 2024-06-21 19:21:00 +00:00 · 2024-06-21 19:21:00 +00:00 · ba65d369db
commit ba65d369db
parent 7aa940f20d
1 changed files with 1 additions and 0 deletions
--- a/scripts/train.py
+++ b/scripts/train.py
@ -248,6 +248,7 @@ def main():
                with Timer("move data") as move_data_t:
                    x = batch.pop("video").to(device, dtype)  # [B, C, T, H, W]
                    y = batch.pop("text")
+                timer_list.append(move_data_t)

                # == visual and text encoding ==
                with Timer("encode") as encode_t: