提交 df5d1624 编写于 作者: Y yuchaojie

move transformer net to new modelzoo directory

上级 0e3dfdd0
...@@ -318,7 +318,7 @@ In order to facilitate developers to enjoy the benefits of MindSpore framework a ...@@ -318,7 +318,7 @@ In order to facilitate developers to enjoy the benefits of MindSpore framework a
| Loss | 2.8 | | Loss | 2.8 |
| Params (M) | 213.7 | | Params (M) | 213.7 |
| Checkpoint for inference | 2.4G (.ckpt file) | | Checkpoint for inference | 2.4G (.ckpt file) |
| Scripts | https://gitee.com/mindspore/mindspore/tree/master/model_zoo/Transformer | | Scripts | https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/nlp/transformer |
#### License #### License
......
...@@ -25,12 +25,12 @@ from mindspore.train.model import Model ...@@ -25,12 +25,12 @@ from mindspore.train.model import Model
from mindspore.train.loss_scale_manager import DynamicLossScaleManager from mindspore.train.loss_scale_manager import DynamicLossScaleManager
from mindspore.train.callback import Callback from mindspore.train.callback import Callback
from mindspore import context from mindspore import context
from model_zoo.Transformer.src.transformer_model import TransformerConfig from model_zoo.official.nlp.transformer.src.transformer_model import TransformerConfig
from model_zoo.Transformer.src.transformer_for_train import TransformerNetworkWithLoss, \ from model_zoo.official.nlp.transformer.src.transformer_for_train import TransformerNetworkWithLoss, \
TransformerTrainOneStepWithLossScaleCell TransformerTrainOneStepWithLossScaleCell
from model_zoo.Transformer.src.config import cfg from model_zoo.official.nlp.transformer.src.config import cfg
from model_zoo.Transformer.src.dataset import create_transformer_dataset from model_zoo.official.nlp.transformer.src.dataset import create_transformer_dataset
from model_zoo.Transformer.src.lr_schedule import create_dynamic_lr from model_zoo.official.nlp.transformer.src.lr_schedule import create_dynamic_lr
DATA_DIR = ["/home/workspace/mindspore_dataset/transformer/test-mindrecord"] DATA_DIR = ["/home/workspace/mindspore_dataset/transformer/test-mindrecord"]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册