Skip to content

Commit 2e47608

Browse files
committed
Fix validation set shuffling (#481)
1 parent 0fa09b6 commit 2e47608

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

bayesflow/workflows/basic_workflow.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -923,7 +923,9 @@ def _fit(
923923
elif isinstance(validation_data, int):
924924
raise ValueError(f"No simulator found for generating {validation_data} data sets.")
925925

926-
validation_data = OfflineDataset(data=validation_data, batch_size=dataset.batch_size, adapter=self.adapter)
926+
validation_data = OfflineDataset(
927+
data=validation_data, batch_size=dataset.batch_size, adapter=self.adapter, stage="validation"
928+
)
927929
monitor = "val_loss"
928930
else:
929931
monitor = "loss"
@@ -982,6 +984,8 @@ def _on_training_finished(self):
982984
else:
983985
file_ext = self.checkpoint_name + ".keras"
984986

985-
logging.info(f"""Training is now finished.
987+
logging.info(
988+
f"""Training is now finished.
986989
You can find the trained approximator at '{self.checkpoint_filepath}/{self.checkpoint_name}.{file_ext}'.
987-
To load it, use approximator = keras.saving.load_model(...).""")
990+
To load it, use approximator = keras.saving.load_model(...)."""
991+
)

0 commit comments

Comments
 (0)