提交 f45466ae 编写于 作者: W wuzewu

Set default value for RunConfig in finetune api

上级 41371ec7
...@@ -25,6 +25,8 @@ import numpy as np ...@@ -25,6 +25,8 @@ import numpy as np
from visualdl import LogWriter from visualdl import LogWriter
from paddlehub.common.logger import logger from paddlehub.common.logger import logger
from paddlehub.common.utils import mkdir
from paddlehub.finetune.config import RunConfig
from paddlehub.finetune.strategy import AdamWeightDecayStrategy, DefaultStrategy from paddlehub.finetune.strategy import AdamWeightDecayStrategy, DefaultStrategy
from paddlehub.finetune.checkpoint import load_checkpoint, save_checkpoint from paddlehub.finetune.checkpoint import load_checkpoint, save_checkpoint
from paddlehub.finetune.evaluate import evaluate_cls_task, evaluate_seq_label_task from paddlehub.finetune.evaluate import evaluate_cls_task, evaluate_seq_label_task
...@@ -294,6 +296,12 @@ def _finetune_cls_task(task, data_reader, feed_list, config=None, ...@@ -294,6 +296,12 @@ def _finetune_cls_task(task, data_reader, feed_list, config=None,
def finetune_and_eval(task, data_reader, feed_list, config=None): def finetune_and_eval(task, data_reader, feed_list, config=None):
if config is None:
config = RunConfig()
if not os.path.exists(config.checkpoint_dir):
mkdir(config.checkpoint_dir)
if task.task_type == "sequence_labeling": if task.task_type == "sequence_labeling":
_finetune_seq_label_task( _finetune_seq_label_task(
task, data_reader, feed_list, config, do_eval=True) task, data_reader, feed_list, config, do_eval=True)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册