update test

2025-07-07 02:36:29 -07:00
parent 2276ad7d51
commit 445aaac585
2 changed files with 4 additions and 4 deletions
@@ -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 = []
@@ -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 = []