hotfix exp

2026-05-21 11:59:01 +02:00 · 2024-04-19 18:15:06 +08:00 · 2024-04-19 18:15:06 +08:00 · f6574f1509
commit f6574f1509
parent 74aade586a
1 changed files with 0 additions and 1 deletions
--- a/scripts/train.py
+++ b/scripts/train.py
@ -229,7 +229,6 @@ def main():
        ) as pbar:
            for step, batch in pbar:
                x = batch.pop("video").to(device, dtype)  # [B, C, T, H, W]
-                print(x.shape)
                y = batch.pop("text")
                # Visual and text encoding
                with torch.no_grad():