Skip to content

Commit bbea5f9

Browse files
sdaultonmeta-codesync[bot]
authored andcommitted
Open-source TransferLearningAdapter and register as Generators.BOTL (#5052)
Summary: Pull Request resolved: #5052 Reviewed By: lena-kashtelyan Differential Revision: D96362191 fbshipit-source-id: 0ba5218814504678793a0a10b4af131c8b559db7
1 parent 5cf4b95 commit bbea5f9

File tree

3 files changed

+773
-0
lines changed

3 files changed

+773
-0
lines changed

ax/adapter/registry.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -463,6 +463,7 @@ class Generators(GeneratorRegistryBase):
463463
IN_SAMPLE_UNIFORM = "InSampleUniform"
464464
ST_MTGP = "ST_MTGP"
465465
BO_MIXED = "BO_MIXED"
466+
BOTL = "BOTL"
466467

467468

468469
def _extract_generator_state_after_gen(

ax/adapter/transfer_learning/__init__.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,3 +4,11 @@
44
# LICENSE file in the root directory of this source tree.
55

66
# pyre-strict
7+
8+
from ax.adapter.transfer_learning.adapter import ( # noqa F401
9+
logger,
10+
TARGET_TASK_VALUE,
11+
TL_EXP,
12+
transfer_learning_generator_specs_constructor,
13+
TransferLearningAdapter,
14+
)

0 commit comments

Comments
 (0)