Skip to content

Commit

Permalink
make SchedulerOptions specifiable by the user
Browse files Browse the repository at this point in the history
Summary: Allow users to specify their own SchedulerOptions when submitting AxSweeps

Reviewed By: lena-kashtelyan

Differential Revision: D30139748

fbshipit-source-id: f6a8a787de0a2bcd921c5a2fd4a78c7e3f8a2632
  • Loading branch information
bernardbeckerman authored and facebook-github-bot committed Aug 10, 2021
1 parent 7b0a489 commit 52974a2
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions ax/service/scheduler.py
Original file line number Diff line number Diff line change
Expand Up @@ -179,14 +179,15 @@ class SchedulerOptions:
the minimum of ``self.poll_available_capacity()`` and the number
of generator runs that the generation strategy is able to produce
without more data or reaching its allowed max paralellism limit.
debug_log_run_metadata: Whether to log run_metadata for debugging purposes.
early_stopping_strategy: A ``BaseEarlyStoppingStrategy`` that determines
whether a trial should be stopped given the current state of
the experiment. Used in ``should_stop_trials_early``.
suppress_storage_errors_after_retries: Whether to fully suppress SQL
storage-related errors if encounted, after retrying the call
multiple times. Only use if SQL storage is not important for the given
use case, since this will only log, but not raise, an exception if
its encountered while saving to DB or loading from it.
it's encountered while saving to DB or loading from it.
"""

trial_type: Type[BaseTrial] = Trial
Expand Down Expand Up @@ -1081,7 +1082,6 @@ def _complete_optimization(self, num_preexisting_trials: int) -> Dict[str, Any]:
"""Conclude optimization with waiting for anymore running trials and
return final results via `wait_for_completed_trials_and_report_results`.
"""
print("HERE!!")
self._record_optimization_complete_message()
res = self.wait_for_completed_trials_and_report_results()
# raise an error if the failure rate exceeds tolerance at the end of the sweep
Expand Down

0 comments on commit 52974a2

Please sign in to comment.