We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 481ea60 commit 7bea7f2Copy full SHA for 7bea7f2
1 file changed
recipes/esm2_accelerate/test_train.py
@@ -294,7 +294,6 @@ def test_accelerate_launch_multi_gpu(accelerate_config, model_tag, tmp_path):
294
combined_output = result.stdout + result.stderr
295
try:
296
final_train_loss = extract_final_train_loss(combined_output)
297
- breakpoint()
298
print(f"Final train_loss: {final_train_loss}")
299
assert final_train_loss < 3.0, f"Final train_loss {final_train_loss} should be less than 3.0"
300
except ValueError as e:
0 commit comments