Skip to content

Commit 7bea7f2

Browse files
committed
remove breakpoint in multi-gpu code
Signed-off-by: Peter St. John <pstjohn@nvidia.com>
1 parent 481ea60 commit 7bea7f2

1 file changed

Lines changed: 0 additions & 1 deletion

File tree

recipes/esm2_accelerate/test_train.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -294,7 +294,6 @@ def test_accelerate_launch_multi_gpu(accelerate_config, model_tag, tmp_path):
294294
combined_output = result.stdout + result.stderr
295295
try:
296296
final_train_loss = extract_final_train_loss(combined_output)
297-
breakpoint()
298297
print(f"Final train_loss: {final_train_loss}")
299298
assert final_train_loss < 3.0, f"Final train_loss {final_train_loss} should be less than 3.0"
300299
except ValueError as e:

0 commit comments

Comments
 (0)