diff --git a/autoaug/autoaugment_learners/GenLearner.py b/autoaug/autoaugment_learners/GenLearner.py index b3d70720a964db16fc6577fe588043b480ceb35e..3f4a40234285369f775331eb5ae5e913845a7bf5 100644 --- a/autoaug/autoaugment_learners/GenLearner.py +++ b/autoaug/autoaugment_learners/GenLearner.py @@ -20,6 +20,7 @@ class Genetic_learner(AaLearner): early_stop_num=20, batch_size=8, toy_size=1, + # GenLearner specific settings num_offspring=1, ): diff --git a/autoaug/autoaugment_learners/__init__.py b/autoaug/autoaugment_learners/__init__.py index 1bd747a85253380df0f1e975e8297e10d0602cb3..a85652c39fb68ba33dccba978fa0d0b2a93008eb 100644 --- a/autoaug/autoaugment_learners/__init__.py +++ b/autoaug/autoaugment_learners/__init__.py @@ -1,5 +1,6 @@ from .AaLearner import * from .RsLearner import * +from .GenLearner import * from .GruLearner import * from .EvoLearner import * from .UcbLearner import * \ No newline at end of file diff --git a/docs/source/reference/aa_learners/autoaug.autoaugment_learners.GenLearner.rst b/docs/source/reference/aa_learners/autoaug.autoaugment_learners.GenLearner.rst new file mode 100644 index 0000000000000000000000000000000000000000..e7ae1826f1a56707a4159115b162f75c62a52963 --- /dev/null +++ b/docs/source/reference/aa_learners/autoaug.autoaugment_learners.GenLearner.rst @@ -0,0 +1,7 @@ +:mod:`autoaug.autoaugment_learners`.GenLearner +============================================== + +.. currentmodule:: autoaug.autoaugment_learners + +.. autoclass:: GenLearner + :members: \ No newline at end of file diff --git a/docs/source/reference/autoaugment_learners.rst b/docs/source/reference/autoaugment_learners.rst index 3cd0dbc23c67c6c722a03debf58960e792f44d1d..380eb47c8ac5a2d3a941b43c8422790573ab2175 100644 --- a/docs/source/reference/autoaugment_learners.rst +++ b/docs/source/reference/autoaugment_learners.rst @@ -3,10 +3,11 @@ AutoAugment learners .. toctree:: :maxdepth: 3 - :caption: autoaugment_learners + :caption: AutoAugment Learners aa_learners/autoaug.autoaugment_learners.AaLearner aa_learners/autoaug.autoaugment_learners.EvoLearner + aa_learners/autoaug.autoaugment_learners.GenLearner aa_learners/autoaug.autoaugment_learners.GruLearner aa_learners/autoaug.autoaugment_learners.RsLearner aa_learners/autoaug.autoaugment_learners.UcbLearner \ No newline at end of file