From 445aaac585be8a9b398361a9de3d8d4959b4f613 Mon Sep 17 00:00:00 2001 From: SolitaryThinker Date: Mon, 7 Jul 2025 02:36:29 -0700 Subject: [PATCH] update test --- fastvideo/v1/tests/training/VSA/test_training_loss_VSA.py | 2 +- fastvideo/v1/tests/training/Vanilla/test_training_loss.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/fastvideo/v1/tests/training/VSA/test_training_loss_VSA.py b/fastvideo/v1/tests/training/VSA/test_training_loss_VSA.py index 670cca47c..f8ef11b3b 100644 --- a/fastvideo/v1/tests/training/VSA/test_training_loss_VSA.py +++ b/fastvideo/v1/tests/training/VSA/test_training_loss_VSA.py @@ -111,7 +111,7 @@ def test_distributed_training(): 'avg_step_time': 1.0, 'grad_norm': 0.1, 'step_time': 1.0, - 'train_loss': 0.001 + 'train_loss': 0.01 } failures = [] diff --git a/fastvideo/v1/tests/training/Vanilla/test_training_loss.py b/fastvideo/v1/tests/training/Vanilla/test_training_loss.py index 7f1459cc0..7b4c81a2d 100644 --- a/fastvideo/v1/tests/training/Vanilla/test_training_loss.py +++ b/fastvideo/v1/tests/training/Vanilla/test_training_loss.py @@ -121,10 +121,10 @@ def test_distributed_training(): wandb_summary = json.load(open(summary_file)) fields_and_thresholds = { - 'avg_step_time': 6.0, + 'avg_step_time': 3.0, 'grad_norm': 0.3, - 'step_time': 6.0, - 'train_loss': 0.0025 + 'step_time': 3.0, + 'train_loss': 0.01 } failures = []