Skip to content

Commit

Permalink
Less loss required
Browse files Browse the repository at this point in the history
  • Loading branch information
gongy committed Feb 15, 2024
1 parent 6428d85 commit 74db36c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion ci/check_loss.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@
val_loss = float(results["ValidationLoss"].iloc[-1])

# Arbitrary threshold
max_loss = 0.8 if b"Mixtral" in contents else 0.25
max_loss = 0.25

print(f"Loss: {train_loss:.2f} (training), {val_loss:.2f} (validation)")
sys.exit(val_loss > max_loss)

0 comments on commit 74db36c

Please sign in to comment.