Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleSeg
提交
d4d3156d
P
PaddleSeg
项目概览
PaddlePaddle
/
PaddleSeg
通知
285
Star
8
Fork
1
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
53
列表
看板
标记
里程碑
合并请求
3
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleSeg
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
53
Issue
53
列表
看板
标记
里程碑
合并请求
3
合并请求
3
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
d4d3156d
编写于
8月 30, 2019
作者:
Z
Zeyu Chen
提交者:
GitHub
8月 30, 2019
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Update usage.md
上级
46192e45
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
7 addition
and
5 deletion
+7
-5
docs/usage.md
docs/usage.md
+7
-5
未找到文件。
docs/usage.md
浏览文件 @
d4d3156d
...
...
@@ -45,7 +45,7 @@ python pdseg/export_model.py ${FLAGS} ${OPTIONS}
### 准备工作
在开始教程前,请先确认准备工作已经完成:
1.
下载合适版本的paddlep
addle
1.
正确安装了PaddleP
addle
2.
PaddleSeg相关依赖已经安装
如果有不确认的地方,请参考
[
安装说明
](
./installation.md
)
...
...
@@ -65,15 +65,16 @@ wget https://paddleseg.bj.bcebos.com/dataset/mini_pet.zip --no-check-certificate
unzip mini_pet.zip
```
### Finetune
接着开始Finetune,为了方便体验,我们在configs目录下放置了Oxford-IIIT Pet所对应的配置文件
`unet_pet.yaml`
,可以通过
`--cfg`
指向该文件来设置训练配置。
### 模型训练
为了方便体验,我们在configs目录下放置了Oxford-IIIT Pet所对应的配置文件
`unet_pet.yaml`
,可以通过
`--cfg`
指向该文件来设置训练配置。
我们选择两张GPU进行训练,这可以通过环境变量
`CUDA_VISIBLE_DEVICES`
来指定。
除此之外,我们指定总BATCH_SIZE为4,PaddleSeg会根据可用的GPU数量,将数据平分到每张卡上,务必确保BATCH_SIZE为GPU数量的整数倍(在本例中,每张卡的BATCH_SIZE为2)。
**NOTE**
: PaddleSeg会根据可用的GPU数量,将Batch内的数据平分到每张卡上,务必确保BATCH_SIZE为训练GPU数量的整数倍
```
export CUDA_VISIBLE_DEVICES=0
,1
export CUDA_VISIBLE_DEVICES=0
python pdseg/train.py --use_gpu \
--do_eval \
--use_tb \
...
...
@@ -101,6 +102,7 @@ python pdseg/train.py --use_gpu \
### 训练过程可视化
当打开do_eval和use_tb两个开关后,我们可以通过TensorBoard查看训练的效果
```
shell
tensorboard
--logdir
train_log
--host
{
$HOST_IP
}
--port
{
$PORT
}
```
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录