move transformer net to new modelzoo directory
This commit is contained in:
parent
0e3dfdd01c
commit
df5d162417
|
@ -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
|
||||||
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"]
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue