From ba65d369db5e79140fb0be71fefe8255a73663b7 Mon Sep 17 00:00:00 2001
From: zhengzangw <zhengzangw@gmail.com>
Date: Fri, 21 Jun 2024 19:21:00 +0000
Subject: [PATCH] [fix] time list

---
 scripts/train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/scripts/train.py b/scripts/train.py
index 574fc0b..98bb2d7 100644
--- 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: