Add config for MedianStoppingRule

README.md

@@ -31,7 +31,7 @@ SIATune is an open-source deep learning model hyperparameter tuning toolbox espe
   Various scheduling techniques are supported to efficiently manage many experiments.
   - [x] [Asynchronous HyperBand](https://arxiv.org/abs/1810.05934)
   - [x] [HyperBand](https://arxiv.org/abs/1603.06560)
-  - [ ] [Median Stopping Rule](https://research.google.com/pubs/pub46180.html)
+  - [x] [Median Stopping Rule](https://research.google.com/pubs/pub46180.html)
   - [ ] [Population Based Training](https://www.deepmind.com/blog/population-based-training-of-neural-networks)
   - [ ] [Population Based Bandits](https://arxiv.org/abs/2002.02518)
   - [x] [Bayesian Optimization and HyperBand](https://arxiv.org/abs/1807.01774)

configs/_base_/scheduler/median.py

@@ -0,0 +1,9 @@
+# Refer to https://github.com/ray-project/ray/blob/master/python/ray/tune/schedulers/median_stopping_rule.py  # noqa
+
+trial_scheduler = dict(
+    type='MedianStoppingRule',
+    time_attr='time_total_s',
+    grace_period=60,
+    min_samples_required=3,
+    min_time_slice=0,
+    hard_stop=True)