Browse Source

!3195 move transformer net to new modelzoo directory

Merge pull request !3195 from yuchaojie/transformer_st
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
c5eae497be
21 changed files with 7 additions and 7 deletions
  1. +1
    -1
      model_zoo/README.md
  2. +0
    -0
      model_zoo/official/nlp/transformer/README.md
  3. +0
    -0
      model_zoo/official/nlp/transformer/create_data.py
  4. +0
    -0
      model_zoo/official/nlp/transformer/eval.py
  5. +0
    -0
      model_zoo/official/nlp/transformer/scripts/process_output.sh
  6. +0
    -0
      model_zoo/official/nlp/transformer/scripts/replace-quote.perl
  7. +0
    -0
      model_zoo/official/nlp/transformer/scripts/run_distribute_train.sh
  8. +0
    -0
      model_zoo/official/nlp/transformer/scripts/run_standalone_train.sh
  9. +0
    -0
      model_zoo/official/nlp/transformer/src/__init__.py
  10. +0
    -0
      model_zoo/official/nlp/transformer/src/beam_search.py
  11. +0
    -0
      model_zoo/official/nlp/transformer/src/config.py
  12. +0
    -0
      model_zoo/official/nlp/transformer/src/dataset.py
  13. +0
    -0
      model_zoo/official/nlp/transformer/src/eval_config.py
  14. +0
    -0
      model_zoo/official/nlp/transformer/src/lr_schedule.py
  15. +0
    -0
      model_zoo/official/nlp/transformer/src/process_output.py
  16. +0
    -0
      model_zoo/official/nlp/transformer/src/tokenization.py
  17. +0
    -0
      model_zoo/official/nlp/transformer/src/transformer_for_train.py
  18. +0
    -0
      model_zoo/official/nlp/transformer/src/transformer_model.py
  19. +0
    -0
      model_zoo/official/nlp/transformer/src/weight_init.py
  20. +0
    -0
      model_zoo/official/nlp/transformer/train.py
  21. +6
    -6
      tests/st/model_zoo_tests/transformer/test_transformer.py

+ 1
- 1
model_zoo/README.md View File

@@ -318,7 +318,7 @@ In order to facilitate developers to enjoy the benefits of MindSpore framework a
| Loss | 2.8 |
| Params (M) | 213.7 |
| 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


model_zoo/Transformer/README.md → model_zoo/official/nlp/transformer/README.md View File


model_zoo/Transformer/create_data.py → model_zoo/official/nlp/transformer/create_data.py View File


model_zoo/Transformer/eval.py → model_zoo/official/nlp/transformer/eval.py View File


model_zoo/Transformer/scripts/process_output.sh → model_zoo/official/nlp/transformer/scripts/process_output.sh View File


model_zoo/Transformer/scripts/replace-quote.perl → model_zoo/official/nlp/transformer/scripts/replace-quote.perl View File


model_zoo/Transformer/scripts/run_distribute_train.sh → model_zoo/official/nlp/transformer/scripts/run_distribute_train.sh View File


model_zoo/Transformer/scripts/run_standalone_train.sh → model_zoo/official/nlp/transformer/scripts/run_standalone_train.sh View File


model_zoo/Transformer/src/__init__.py → model_zoo/official/nlp/transformer/src/__init__.py View File


model_zoo/Transformer/src/beam_search.py → model_zoo/official/nlp/transformer/src/beam_search.py View File


model_zoo/Transformer/src/config.py → model_zoo/official/nlp/transformer/src/config.py View File


model_zoo/Transformer/src/dataset.py → model_zoo/official/nlp/transformer/src/dataset.py View File


model_zoo/Transformer/src/eval_config.py → model_zoo/official/nlp/transformer/src/eval_config.py View File


model_zoo/Transformer/src/lr_schedule.py → model_zoo/official/nlp/transformer/src/lr_schedule.py View File


model_zoo/Transformer/src/process_output.py → model_zoo/official/nlp/transformer/src/process_output.py View File


model_zoo/Transformer/src/tokenization.py → model_zoo/official/nlp/transformer/src/tokenization.py View File


model_zoo/Transformer/src/transformer_for_train.py → model_zoo/official/nlp/transformer/src/transformer_for_train.py View File


model_zoo/Transformer/src/transformer_model.py → model_zoo/official/nlp/transformer/src/transformer_model.py View File


model_zoo/Transformer/src/weight_init.py → model_zoo/official/nlp/transformer/src/weight_init.py View File


model_zoo/Transformer/train.py → model_zoo/official/nlp/transformer/train.py View File


+ 6
- 6
tests/st/model_zoo_tests/transformer/test_transformer.py View File

@@ -25,12 +25,12 @@ from mindspore.train.model import Model
from mindspore.train.loss_scale_manager import DynamicLossScaleManager
from mindspore.train.callback import Callback
from mindspore import context
from model_zoo.Transformer.src.transformer_model import TransformerConfig
from model_zoo.Transformer.src.transformer_for_train import TransformerNetworkWithLoss, \
TransformerTrainOneStepWithLossScaleCell
from model_zoo.Transformer.src.config import cfg
from model_zoo.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.transformer_model import TransformerConfig
from model_zoo.official.nlp.transformer.src.transformer_for_train import TransformerNetworkWithLoss, \
TransformerTrainOneStepWithLossScaleCell
from model_zoo.official.nlp.transformer.src.config import cfg
from model_zoo.official.nlp.transformer.src.dataset import create_transformer_dataset
from model_zoo.official.nlp.transformer.src.lr_schedule import create_dynamic_lr

DATA_DIR = ["/home/workspace/mindspore_dataset/transformer/test-mindrecord"]



Loading…
Cancel
Save