diff --git a/run/entity_extraction/tener/__init__.py b/run/entity_extraction/adaptiveTransformerNER/__init__.py similarity index 100% rename from run/entity_extraction/tener/__init__.py rename to run/entity_extraction/adaptiveTransformerNER/__init__.py diff --git a/run/entity_extraction/tener/modules/TransformerEmbedding.py b/run/entity_extraction/adaptiveTransformerNER/modules/TransformerEmbedding.py similarity index 100% rename from run/entity_extraction/tener/modules/TransformerEmbedding.py rename to run/entity_extraction/adaptiveTransformerNER/modules/TransformerEmbedding.py diff --git a/run/entity_extraction/tener/modules/__init__.py b/run/entity_extraction/adaptiveTransformerNER/modules/__init__.py similarity index 100% rename from run/entity_extraction/tener/modules/__init__.py rename to run/entity_extraction/adaptiveTransformerNER/modules/__init__.py diff --git a/run/entity_extraction/tener/modules/callbacks.py b/run/entity_extraction/adaptiveTransformerNER/modules/callbacks.py similarity index 100% rename from run/entity_extraction/tener/modules/callbacks.py rename to run/entity_extraction/adaptiveTransformerNER/modules/callbacks.py diff --git a/run/entity_extraction/tener/modules/pipe.py b/run/entity_extraction/adaptiveTransformerNER/modules/pipe.py similarity index 100% rename from run/entity_extraction/tener/modules/pipe.py rename to run/entity_extraction/adaptiveTransformerNER/modules/pipe.py diff --git a/run/entity_extraction/tener/modules/relative_transformer.py b/run/entity_extraction/adaptiveTransformerNER/modules/relative_transformer.py similarity index 100% rename from run/entity_extraction/tener/modules/relative_transformer.py rename to run/entity_extraction/adaptiveTransformerNER/modules/relative_transformer.py diff --git a/run/entity_extraction/tener/modules/transformer.py b/run/entity_extraction/adaptiveTransformerNER/modules/transformer.py similarity index 100% rename from run/entity_extraction/tener/modules/transformer.py rename to run/entity_extraction/adaptiveTransformerNER/modules/transformer.py diff --git a/run/entity_extraction/tener/modules/utils.py b/run/entity_extraction/adaptiveTransformerNER/modules/utils.py similarity index 100% rename from run/entity_extraction/tener/modules/utils.py rename to run/entity_extraction/adaptiveTransformerNER/modules/utils.py diff --git a/run/entity_extraction/tener/train_tener_cn.py b/run/entity_extraction/adaptiveTransformerNER/train_tener_cn.py similarity index 96% rename from run/entity_extraction/tener/train_tener_cn.py rename to run/entity_extraction/adaptiveTransformerNER/train_tener_cn.py index 06028d5..bafe2f6 100644 --- a/run/entity_extraction/tener/train_tener_cn.py +++ b/run/entity_extraction/adaptiveTransformerNER/train_tener_cn.py @@ -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()