ViperEkura/AstrAI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: 修复参数传递问题

Browse Source
This commit is contained in:
ViperEkura
2026-03-30 22:22:36 +08:00
parent e1f9901384
commit 345fd2f091
6 changed files with 13 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/trainer/test_trainer.py
+6 -3
View File
@@ -30,7 +30,8 @@ def test_different_batch_sizes(base_test_env, random_dataset):
checkpoint_interval=5,
accumulation_steps=1,
max_grad_norm=1.0,
random_seed=np.random.randint(1000)
random_seed=np.random.randint(1000),
device_type=base_test_env["device"]
)
assert train_config.batch_size == batch_size
@@ -59,7 +60,8 @@ def test_gradient_accumulation(base_test_env, random_dataset):
checkpoint_interval=10,
accumulation_steps=accumulation_steps,
max_grad_norm=1.0,
random_seed=42
random_seed=42,
device_type=base_test_env["device"]
)
trainer = Trainer(train_config)
@@ -96,7 +98,8 @@ def test_memory_efficient_training(base_test_env, random_dataset):
checkpoint_interval=5,
accumulation_steps=config["accumulation_steps"],
max_grad_norm=1.0,
random_seed=42
random_seed=42,
device_type=base_test_env["device"]
)
assert train_config.accumulation_steps == config["accumulation_steps"]