fix: 修复参数传递问题

2026-03-30 22:22:36 +08:00
parent e1f9901384
commit 345fd2f091
6 changed files with 13 additions and 5 deletions
@@ -25,7 +25,8 @@ def test_callback_integration(base_test_env, random_dataset):
        checkpoint_interval=3,
        accumulation_steps=1,
        max_grad_norm=1.0,
-        random_seed=42
+        random_seed=42,
+        device_type=base_test_env["device"]
    )