Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
mobilenet_prune_pytorch
Commits
329669e5
Commit
329669e5
authored
Jan 04, 2024
by
dcuai
Browse files
Update README.md
parent
0f7c143d
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
24 additions
and
14 deletions
+24
-14
README.md
README.md
+24
-14
No files found.
README.md
View file @
329669e5
# mobilenet_prune
## 论文
EfficientNet
https://arxiv.org/pdf/1905.11946.pdf
mobilenetv2
https://arxiv.org/pdf/1801.04381.pdf
mobilenetv3
https://arxiv.org/pdf/1905.02244.pdf
*
EfficientNet https://arxiv.org/pdf/1905.11946.pdf
*
mobilenetv2 https://arxiv.org/pdf/1801.04381.pdf
*
mobilenetv3 https://arxiv.org/pdf/1905.02244.pdf
## 模型结构
mobilenetv2:

详细模型结构见官方论文:
EfficientNet
*
https://arxiv.org/pdf/1905.11946.pdf
mobilenetv2
*
https://arxiv.org/pdf/1801.04381.pdf
mobilenetv3
*
https://arxiv.org/pdf/1905.02244.pdf
*
EfficientNet https://arxiv.org/pdf/1905.11946.pdf
*
mobilenetv2 https://arxiv.org/pdf/1801.04381.pdf
*
mobilenetv3 https://arxiv.org/pdf/1905.02244.pdf
## 算法原理
CNN原理
## 数据集
torchvision.datasets.CIFAR10
`CIFAR-10`
-
[
https://www.cs.toronto.edu/~kriz/cifar.html
](
https://www.cs.toronto.edu/~kriz/cifar.html
)
训练数据目录结构如下,用于正常训练的完整数据集请按此目录结构进行制备:
```
├── cifar-10-batches-py
│ ├── batches.meta
│ ├── data_batch_1
│ ├── data_batch_2
│ ├── data_batch_3
│ ├── data_batch_4
│ ├── data_batch_5
│ ├── readme.html
│ └── test_batch
```
**
tips:本模型也会自动下载数据集
## 环境配置
在
[
光源
](
https://sourcefind.cn/#/main-page
)
可拉取剪枝及微调的docker镜像,在
[
光合开发者社区
](
https://cancon.hpccube.com:65024/4/main/
)
可下载torch安装包。mobilenet_prune推荐的镜像如下:
```
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment