提交 fec9c009 编写于 作者: W wangxiao1021

fix bugs

上级 7a47bd0a
import downloader from . import downloader
# from mtl_controller import Controller # from mtl_controller import Controller
#import controller #import controller
import optimizer from . import optimizer
import lr_sched from . import lr_sched
import backbone from . import backbone
import reader from . import reader
import head from . import head
from trainer import Trainer from .trainer import Trainer
from multihead_trainer import MultiHeadTrainer from .multihead_trainer import MultiHeadTrainer
#del interface #del interface
#del task_instance #del task_instance
......
from ernie import ERNIE from .ernie import ERNIE
from bert import BERT from .bert import BERT
...@@ -5,5 +5,5 @@ import multiprocessing ...@@ -5,5 +5,5 @@ import multiprocessing
gpu_dev_count = int(fluid.core.get_cuda_device_count()) gpu_dev_count = int(fluid.core.get_cuda_device_count())
cpu_dev_count = int(os.environ.get('CPU_NUM', multiprocessing.cpu_count())) cpu_dev_count = int(os.environ.get('CPU_NUM', multiprocessing.cpu_count()))
from reader import yield_pieces, data_feeder, decode_fake from .reader import yield_pieces, data_feeder, decode_fake
from _downloader import * from ._downloader import *
\ No newline at end of file
from cls import Classify from .cls import Classify
from match import Match from .match import Match
from ner import SequenceLabel from .ner import SequenceLabel
from mrc import MRC from .mrc import MRC
from mlm import MaskLM from .mlm import MaskLM
from slanted_triangular_schedualer import TriangularSchedualer from .slanted_triangular_schedualer import TriangularSchedualer
from warmup_schedualer import WarmupSchedualer from .warmup_schedualer import WarmupSchedualer
from adam import Adam from .adam import Adam
from cls import ClassifyReader from .cls import ClassifyReader
from match import MatchReader from .match import MatchReader
from seq_label import SequenceLabelReader from .seq_label import SequenceLabelReader
from mrc import MRCReader from .mrc import MRCReader
from mlm import MaskLMReader from .mlm import MaskLMReader
import basic_helper from . import basic_helper
import config_helper from . import config_helper
...@@ -2,7 +2,7 @@ ...@@ -2,7 +2,7 @@
import os import os
import json import json
import yaml import yaml
from config_helper import PDConfig from .config_helper import PDConfig
import logging import logging
from paddle import fluid from paddle import fluid
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册