Skip to content

Commit e0ca830

Browse files
committed
checkstyle
Signed-off-by: Tcc0403 <[email protected]>
1 parent 32619e9 commit e0ca830

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

test/convergence/bf16/test_mini_models.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -932,7 +932,6 @@ def run_mini_model(
932932
print(f"Step {i}, Loss: {output.loss.item()}")
933933
loss_list.append(output.loss.item())
934934

935-
936935
model.eval()
937936
eval_batch = next(loader_iter).to(model.device)
938937
if with_liger:
@@ -1325,7 +1324,6 @@ def test_mini_model(
13251324

13261325
actual_output = run_mini_model(model_name=model_name, num_steps=num_steps, dtype=dtype, lr=lr, with_liger=True)
13271326

1328-
13291327
# Compare every step of the loss
13301328
assert_verbose_allclose(
13311329
torch.tensor([expected_output["loss"]]),

0 commit comments

Comments
 (0)