Skip to content
GitLab
Explore
Sign in
Primary navigation
Search or go to…
Project
2
221cb0332
Manage
Activity
Members
Labels
Plan
Issues
0
Issue boards
Milestones
Wiki
Code
Merge requests
0
Repository
Branches
Commits
Tags
Repository graph
Compare revisions
Snippets
Build
Pipelines
Jobs
Pipeline schedules
Artifacts
Deploy
Releases
Package Registry
Operate
Environments
Terraform modules
Monitor
Incidents
Analyze
Value stream analytics
Contributor analytics
CI/CD analytics
Repository analytics
Help
Help
Support
GitLab documentation
Compare GitLab plans
Community forum
Contribute to GitLab
Provide feedback
Keyboard shortcuts
?
Snippets
Groups
Projects
Show more breadcrumbs
Summer2022
221cb0332
Commits
7c7aad84
Commit
7c7aad84
authored
3 years ago
by
lianghao23
Browse files
Options
Downloads
Patches
Plain Diff
Fix README for CPU, and the name of default config file.
parent
5ee12519
No related branches found
Branches containing commit
No related tags found
Tags containing commit
No related merge requests found
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
official/cv/googlenet/README.md
+6
-2
6 additions, 2 deletions
official/cv/googlenet/README.md
official/cv/googlenet/README_CN.md
+6
-2
6 additions, 2 deletions
official/cv/googlenet/README_CN.md
official/cv/googlenet/model_utils/config.py
+1
-1
1 addition, 1 deletion
official/cv/googlenet/model_utils/config.py
with
13 additions
and
5 deletions
official/cv/googlenet/README.md
+
6
−
2
View file @
7c7aad84
...
...
@@ -125,7 +125,7 @@ After installing MindSpore via the official website, you can start training and
-
running on GPU
For running on GPU, please change
`device_target`
from
`Ascend`
to
`GPU`
in configuration file
src/
config.
p
y
For running on GPU, please change
`device_target`
from
`Ascend`
to
`GPU`
in configuration file
[dataset]_
config.y
aml
```
python
# run training example
...
...
@@ -145,10 +145,14 @@ After installing MindSpore via the official website, you can start training and
```
python
# run training example
bash
run_train_cpu
.
sh
OR
nohup
python
train
.
py
--
config_path
=
cifar10_config_cpu
.
yaml
--
dataset_name
=
cifar10
>
train
.
log
2
>&
1
&
# run evaluation example
nohup
python
eval
.
py
--
checkpoint_path
=
[
CHECKPOINT_PATH
]
>
eval
.
log
2
>&
1
&
bash
run_eval_cpu
.
sh
OR
nohup
python
eval
.
py
--
checkpoint_path
=
[
CHECKPOINT_PATH
]
--
config_path
=
cifar10_config_cpu
.
yaml
>
eval
.
log
2
>&
1
&
```
We use CIFAR-10 dataset by default. Your can also pass
`$dataset_type`
to the scripts so that select different datasets. For more details, please refer the specify script.
...
...
This diff is collapsed.
Click to expand it.
official/cv/googlenet/README_CN.md
+
6
−
2
View file @
7c7aad84
...
...
@@ -127,7 +127,7 @@ GoogleNet由多个inception模块串联起来,可以更加深入。 降维的
-
GPU处理器环境运行
为了在GPU处理器环境运行,请将配置文件
src/
config.
p
y中的
`device_target`
从
`Ascend`
改为
`GPU`
为了在GPU处理器环境运行,请将配置文件
[dataset]_
config.y
aml
中的
`device_target`
从
`Ascend`
改为
`GPU`
```
python
# 运行训练示例
...
...
@@ -147,10 +147,14 @@ GoogleNet由多个inception模块串联起来,可以更加深入。 降维的
```
python
# 运行训练示例
bash
run_train_cpu
.
sh
OR
nohup
python
train
.
py
--
config_path
=
cifar10_config_cpu
.
yaml
--
dataset_name
=
cifar10
>
train
.
log
2
>&
1
&
# 运行评估示例
nohup
python
eval
.
py
--
checkpoint_path
=
[
CHECKPOINT_PATH
]
>
eval
.
log
2
>&
1
&
bash
run_eval_cpu
.
sh
OR
nohup
python
eval
.
py
--
checkpoint_path
=
[
CHECKPOINT_PATH
]
--
config_path
=
cifar10_config_cpu
.
yaml
>
eval
.
log
2
>&
1
&
```
默认使用CIFAR-10数据集。您也可以将
`$dataset_type`
传入脚本,以便选择其他数据集。如需查看更多详情,请参考指定脚本。
...
...
This diff is collapsed.
Click to expand it.
official/cv/googlenet/model_utils/config.py
+
1
−
1
View file @
7c7aad84
...
...
@@ -115,7 +115,7 @@ def get_config():
"""
parser
=
argparse
.
ArgumentParser
(
description
=
"
default name
"
,
add_help
=
False
)
current_dir
=
os
.
path
.
dirname
(
os
.
path
.
abspath
(
__file__
))
parser
.
add_argument
(
"
--config_path
"
,
type
=
str
,
default
=
os
.
path
.
join
(
current_dir
,
"
../
default
_config.yaml
"
),
parser
.
add_argument
(
"
--config_path
"
,
type
=
str
,
default
=
os
.
path
.
join
(
current_dir
,
"
../
cifar10
_config.yaml
"
),
help
=
"
Config file path
"
)
path_args
,
_
=
parser
.
parse_known_args
()
default
,
helper
,
choices
=
parse_yaml
(
path_args
.
config_path
)
...
...
This diff is collapsed.
Click to expand it.
Preview
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Save comment
Cancel
Please
register
or
sign in
to comment