add some layers and models

This commit is contained in:
loujie0822 2020-05-12 20:59:43 +08:00
parent 3e2c52478b
commit 74c411c1ec
9 changed files with 2 additions and 2 deletions

View File

@ -7,8 +7,8 @@ from fastNLP import Trainer, GradientClipCallback, WarmupCallback
from fastNLP import cache_results
from fastNLP.embeddings import StaticEmbedding, BertEmbedding
from models.ner_net.tener import TENER
from run.entity_extraction.tener.modules.callbacks import EvaluateCallback
from run.entity_extraction.tener.modules.pipe import CNNERPipe
from run.entity_extraction.adaptiveTransformerNER.modules.callbacks import EvaluateCallback
from run.entity_extraction.adaptiveTransformerNER.modules.pipe import CNNERPipe
device = 'cpu'
parser = argparse.ArgumentParser()