mirror of
https://github.com/microsoft/FLAML.git
synced 2026-02-09 02:09:16 +08:00
set verbose for transformers
This commit is contained in:
@@ -396,12 +396,14 @@ class TransformersEstimator(BaseEstimator):
|
||||
)
|
||||
|
||||
def fit(self, X_train: DataFrame, y_train: Series, budget=None, **kwargs):
|
||||
import transformers
|
||||
|
||||
transformers.logging.set_verbosity_error()
|
||||
|
||||
from transformers import EarlyStoppingCallback
|
||||
from transformers.trainer_utils import set_seed
|
||||
from transformers import AutoTokenizer
|
||||
from transformers.data import DataCollatorWithPadding
|
||||
|
||||
import transformers
|
||||
from datasets import Dataset
|
||||
from .nlp.utils import (
|
||||
get_num_labels,
|
||||
|
||||
Reference in New Issue
Block a user