github/autogen
1
1
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2026-04-20 03:02:16 -04:00
Code Issues Packages Projects Releases Wiki Activity

fixing bug

Browse Source
This commit is contained in:
liususan091219
2022-03-27 20:10:51 -04:00
committed by Chi Wang
parent 6c16e47e42
commit 0bcf618fea
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
flaml/nlp/utils.py
View File

@@ -609,7 +609,7 @@ class HFArgs:
def __post_init__(self):
if self.task in NLG_TASKS:
HFArgs.model_path = "t5-small"
self.model_path = "t5-small"
@staticmethod
def load_args_from_console():

2
notebook/automl_nlp.ipynb
View File

@@ -1398,7 +1398,7 @@
"\n",
"automl_settings = {\n",
" \"time_budget\": 500, # setting the time budget\n",
" \"task\": \"summarization\", # setting the task as multiplechoice-classification\n",
" \"task\": \"summarization\", # setting the task as summarization\n",
" \"hf_args\": {\n",
" \"output_dir\": \"data/output/\", # setting the output directory\n",
" \"ckpt_per_epoch\": 1, # setting the number of checkoints per epoch\n",