Skip to content

Commit 9c25c73

Browse files
committed
feat: deleted embedder config
1 parent d57be55 commit 9c25c73

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

autointent/generation/utterances/evolution/incremental_evolver.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@
1313
from datasets import concatenate_datasets
1414

1515
from autointent import Dataset, Pipeline
16-
from autointent.configs import EmbedderConfig
1716
from autointent.custom_types import Split
1817
from autointent.generation.utterances import UtteranceEvolver
1918
from autointent.generation.utterances.generator import Generator
@@ -83,8 +82,6 @@ def augment(
8382
8483
Note that for now it supports only single-label datasets.
8584
"""
86-
emb_config = EmbedderConfig(batch_size=16, device="cuda")
87-
8885
best_result = 0
8986
merge_dataset = copy.deepcopy(dataset)
9087

@@ -95,7 +92,6 @@ def augment(
9592
merge_dataset[split_name] = concatenate_datasets([merge_dataset[split_name], new_samples_dataset])
9693

9794
pipeline_optimizer = Pipeline.from_search_space(self.search_space)
98-
pipeline_optimizer.set_config(emb_config)
9995
ctx = pipeline_optimizer.fit(merge_dataset)
10096
results = ctx.optimization_info.dump_evaluation_results()
10197
decision_metric = results["metrics"]["decision"][0]

0 commit comments

Comments
 (0)