Skip to content

Commit

Permalink
Make the default number of cycles in the training example to be 20 as…
Browse files Browse the repository at this point in the history
… users typically let it run fully
  • Loading branch information
golmschenk committed Mar 9, 2024
1 parent 20be387 commit 007a85e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/transit_train.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ def main():
validation_light_curve_dataset = get_transit_validation_dataset()
model = Hadryss.new()
train_hyperparameter_configuration = TrainHyperparameterConfiguration.new(
batch_size=100, cycles=100, train_steps_per_cycle=100, validation_steps_per_cycle=10)
batch_size=100, cycles=20, train_steps_per_cycle=100, validation_steps_per_cycle=10)
train_session(train_datasets=[train_light_curve_dataset], validation_datasets=[validation_light_curve_dataset],
model=model, hyperparameter_configuration=train_hyperparameter_configuration)

Expand Down

0 comments on commit 007a85e

Please sign in to comment.