Browse Source

!3233 change lenet and alexnet dir

Merge pull request !3233 from wukesong/ori_master
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
7bb7328d1d
16 changed files with 3 additions and 3 deletions
  1. +2
    -2
      model_zoo/README.md
  2. +0
    -0
      model_zoo/official/cv/alexnet/README.md
  3. +0
    -0
      model_zoo/official/cv/alexnet/eval.py
  4. +0
    -0
      model_zoo/official/cv/alexnet/src/__init__.py
  5. +1
    -1
      model_zoo/official/cv/alexnet/src/alexnet.py
  6. +0
    -0
      model_zoo/official/cv/alexnet/src/config.py
  7. +0
    -0
      model_zoo/official/cv/alexnet/src/dataset.py
  8. +0
    -0
      model_zoo/official/cv/alexnet/src/generator_lr.py
  9. +0
    -0
      model_zoo/official/cv/alexnet/train.py
  10. +0
    -0
      model_zoo/official/cv/lenet/README.md
  11. +0
    -0
      model_zoo/official/cv/lenet/eval.py
  12. +0
    -0
      model_zoo/official/cv/lenet/src/__init__.py
  13. +0
    -0
      model_zoo/official/cv/lenet/src/config.py
  14. +0
    -0
      model_zoo/official/cv/lenet/src/dataset.py
  15. +0
    -0
      model_zoo/official/cv/lenet/src/lenet.py
  16. +0
    -0
      model_zoo/official/cv/lenet/train.py

+ 2
- 2
model_zoo/README.md View File

@@ -148,7 +148,7 @@ In order to facilitate developers to enjoy the benefits of MindSpore framework a
| Loss | 0.108 | | Loss | 0.108 |
| Params (M) | 61.10 | | Params (M) | 61.10 |
| Checkpoint for Fine tuning | 445MB(.ckpt file) | | Checkpoint for Fine tuning | 445MB(.ckpt file) |
| Scripts | https://gitee.com/mindspore/mindspore/tree/master/model_zoo/alexnet|
| Scripts | https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/alexnet|
#### [LeNet](#table-of-contents) #### [LeNet](#table-of-contents)
@@ -168,7 +168,7 @@ In order to facilitate developers to enjoy the benefits of MindSpore framework a
| Loss | 0.004 | | Loss | 0.004 |
| Params (M) | 0.06 | | Params (M) | 0.06 |
| Checkpoint for Fine tuning | 483KB(.ckpt file) | | Checkpoint for Fine tuning | 483KB(.ckpt file) |
| Scripts | https://gitee.com/mindspore/mindspore/tree/master/model_zoo/lenet|
| Scripts | https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/lenet|
### Object Detection and Segmentation ### Object Detection and Segmentation


model_zoo/alexnet/README.md → model_zoo/official/cv/alexnet/README.md View File


model_zoo/alexnet/eval.py → model_zoo/official/cv/alexnet/eval.py View File


model_zoo/alexnet/src/__init__.py → model_zoo/official/cv/alexnet/src/__init__.py View File


model_zoo/alexnet/src/alexnet.py → model_zoo/official/cv/alexnet/src/alexnet.py View File

@@ -29,7 +29,7 @@ def fc_with_initialize(input_channels, out_channels):
return nn.Dense(input_channels, out_channels, weight, bias) return nn.Dense(input_channels, out_channels, weight, bias)


def weight_variable(): def weight_variable():
return TruncatedNormal(0.02) # 0.02
return TruncatedNormal(0.02)




class AlexNet(nn.Cell): class AlexNet(nn.Cell):

model_zoo/alexnet/src/config.py → model_zoo/official/cv/alexnet/src/config.py View File


model_zoo/alexnet/src/dataset.py → model_zoo/official/cv/alexnet/src/dataset.py View File


model_zoo/alexnet/src/generator_lr.py → model_zoo/official/cv/alexnet/src/generator_lr.py View File


model_zoo/alexnet/train.py → model_zoo/official/cv/alexnet/train.py View File


model_zoo/lenet/README.md → model_zoo/official/cv/lenet/README.md View File


model_zoo/lenet/eval.py → model_zoo/official/cv/lenet/eval.py View File


model_zoo/lenet/src/__init__.py → model_zoo/official/cv/lenet/src/__init__.py View File


model_zoo/lenet/src/config.py → model_zoo/official/cv/lenet/src/config.py View File


model_zoo/lenet/src/dataset.py → model_zoo/official/cv/lenet/src/dataset.py View File


model_zoo/lenet/src/lenet.py → model_zoo/official/cv/lenet/src/lenet.py View File


model_zoo/lenet/train.py → model_zoo/official/cv/lenet/train.py View File


Loading…
Cancel
Save