fix: 修复参数传递问题

2026-03-30 22:22:36 +08:00
parent e1f9901384
commit 345fd2f091
6 changed files with 13 additions and 5 deletions
@@ -25,6 +25,7 @@ def test_early_stopping_simulation(base_test_env, early_stopping_dataset):
        checkpoint_interval=1,
        accumulation_steps=2,
        random_seed=np.random.randint(1e4),
+        device_type=base_test_env["device"]
    )

    trainer = Trainer(train_config)