yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-12-25 05:59:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

set verbose for transformers

Browse Source
This commit is contained in:
liususan091219 2022-01-12 00:10:31 -05:00 committed by Chi Wang
parent 99667dad5f
commit 303d40c76c
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
flaml/model.py
View File

@ -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,