Skip to content
Snippets Groups Projects
Commit acd9a255 authored by yexijoe's avatar yexijoe
Browse files

add eval_ascend

parent 08b16337
No related branches found
No related tags found
No related merge requests found
......@@ -127,8 +127,11 @@
├── README_CN.md // FishNet99相关说明
├── ascend310_infer // 实现310推理源代码
├── scripts
│ ├──run_eval_ascend.sh // Ascend评估的shell脚本
│ ├──run_eval_gpu.sh // GPU评估的shell脚本
│ ├──run_infer_310.sh // Ascend推理的shell脚本
│ ├──run_standalone_train_ascend.sh // Ascend单卡训练的shell脚本
│ ├──run_standalone_train_gpu.sh // GPU单卡训练的shell脚本
│ ├──run_train_ascend.sh // 分布式到Ascend的shell脚本
│ ├──run_train_gpu.sh // 分布式到GPU处理器的shell脚本
├── src
......@@ -145,7 +148,7 @@
在config.py中可以同时配置训练参数和评估参数。
- 配置FishNet99和ImageNet-1k数据集。
- 配置fishnet99和ImageNet-1k数据集。
```python
'name':'imagenet' # 数据集
......@@ -177,6 +180,8 @@
```bash
python train.py --device_id=0 --device_type=Ascend > train.log 2>&1 &
OR
bash ./scripts/run_standalone_train_ascend.sh
```
上述python命令将在后台运行,可以通过生成的train.log文件查看结果。
......@@ -197,6 +202,8 @@
```bash
python train.py --device_id=0 --device_type=GPU > train_gpu.log 2>&1 &
OR
bash ./scripts/run_standalone_train_gpu.sh
```
上述python命令将在后台运行,可以通过生成的train_gpu.log文件查看结果。
......@@ -233,6 +240,8 @@
```bash
python eval.py --checkpoint_path ./ckpt_0 --device_type=Ascend > ./eval.log 2>&1 &
OR
bash ./scripts/run_ascend.sh ./ckpt_0
```
- 在GPU处理器环境运行时评估ImageNet-1k数据集
......@@ -242,7 +251,7 @@
```bash
python eval.py --checkpoint_path ./ckpt_0 --device_type=GPU > ./eval_gpu.log 2>&1 &
OR
bash ./scripts/run_eval.sh
bash ./scripts/run_eval.sh ./ckpt_0
```
## 导出过程
......
#!/bin/bash
# Copyright 2022 Huawei Technologies Co., Ltd
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.mitations under the License.
ckpts_path=$1
python eval.py --checkpoint_path ${ckpts_path} --device_type=Ascend > ./eval_ascend.log 2>&1 &
#!/bin/bash
# Copyright 2022 Huawei Technologies Co., Ltd
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.mitations under the License.
python train.py --device_id=0 --device_type=Ascend > train.log 2>&1 &
#!/bin/bash
# Copyright 2022 Huawei Technologies Co., Ltd
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.mitations under the License.
python train.py --device_id=0 --device_type=GPU > train_gpu.log 2>&1 &
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment