Skip to content

Commit

Permalink
change validation to 80%
Browse files Browse the repository at this point in the history
  • Loading branch information
Radonirinaunimi committed Sep 13, 2024
1 parent d2feade commit 92fec97
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion n3fit/src/n3fit/hyper_optimization/rewards.py
Original file line number Diff line number Diff line change
Expand Up @@ -268,7 +268,7 @@ def compute_loss(
### Experiment:
# Use the validation loss as the loss
# summed with how far from 2 are we for the kfold
validation_loss_average = self.reduce_over_replicas(validation_loss, proportion=0.9)
validation_loss_average = self.reduce_over_replicas(validation_loss, proportion=0.8)
kfold_loss_average = self.reduce_over_replicas(kfold_loss, proportion=0.1)
loss = validation_loss_average + (max(kfold_loss_average, 2.0) - 2.0)
elif self.loss_type == "phi2":
Expand Down

0 comments on commit 92fec97

Please sign in to comment.