未验证 提交 23707ef7 编写于 作者: X Xiaoyao Xi 提交者: GitHub

Merge pull request #52 from wangxiao1021/api

fix install
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
del default_settings #del default_settings
del utils #del utils
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 . import gpu_dev_count, cpu_dev_count from . import gpu_dev_count, cpu_dev_count
import Queue try:
import queue
except ImportError:
import Queue as queue
from threading import Thread from threading import Thread
dev_count = gpu_dev_count if gpu_dev_count > 0 else cpu_dev_count dev_count = gpu_dev_count if gpu_dev_count > 0 else cpu_dev_count
......
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.
先完成此消息的编辑!
想要评论请 注册