Skip to content

Commit 48d9f61

Browse files
[ci] [misc] fix training test threshold (#573)
1 parent 5f938b5 commit 48d9f61

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

fastvideo/v1/tests/training/Vanilla/test_training_loss.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -121,9 +121,9 @@ def test_distributed_training():
121121
wandb_summary = json.load(open(summary_file))
122122

123123
fields_and_thresholds = {
124-
'avg_step_time': 1.0,
124+
'avg_step_time': 6.0,
125125
'grad_norm': 0.3,
126-
'step_time': 0.5,
126+
'step_time': 6.0,
127127
'train_loss': 0.0025
128128
}
129129

0 commit comments

Comments
 (0)