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
MLPerf_RetinaNet_pytorch
Commits
d0ff7db1
Commit
d0ff7db1
authored
Apr 26, 2023
by
liangjing
Browse files
add model.properties
parent
4083247e
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
16 additions
and
3 deletions
+16
-3
README.md
README.md
+7
-3
model.properties
model.properties
+9
-0
No files found.
README.md
View file @
d0ff7db1
...
...
@@ -40,6 +40,10 @@ RetinaNet的网络结构主要分为两个部分:特征提取网络和检测
## 训练
### 测试规模
单机8卡进行性能&&精度测试
### 环境配置
提供
[
光源
](
https://www.sourcefind.cn/#/service-details
)
拉取的训练的docker镜像:
*
训练镜像:
...
...
@@ -53,14 +57,14 @@ python依赖安装:
cd ./cocoapi-0.7.0/PythonAPI; python3 setup.py install
### 训练
训练命令
(此处以单机8卡规模为例说明)
:
训练命令:
nohup bash sbatch.sh >& bs16_epoch6.log &
#输出结果见bs16_epoch6.log
#注:可通过修改dcu.sh中DATASET_DIR参数按需修改输入数据的位置
##
性能和准确率数据
测试
采用上述输入数据,加速卡采用Z100L
,
最终
可
达到收敛
精度
要求
##
测试结果
采用上述输入数据,加速卡采用Z100L
*
8,可
最终达到
官方
收敛要求
;
## 历史版本
*
https://developer.hpccube.com/codes/modelzoo/mlperf_retinanet
...
...
model.properties
0 → 100644
View file @
d0ff7db1
# 模型名称
modelName
=
RetinaNet
# 模型描述
modelDescription
=
RetinaNet是一种基于特征金字塔网络(Feature Pyramid Network)和Focal Loss损失函数的目标检测模型
# 应用场景(多个标签以英文逗号分割)
appScenario
=
CV,Object detection
# 框架类型(多个标签以英文逗号分割)
frameType
=
Pytorch
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