Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleSeg
提交
95b4251e
P
PaddleSeg
项目概览
PaddlePaddle
/
PaddleSeg
通知
286
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看板
未验证
提交
95b4251e
编写于
8月 30, 2019
作者:
W
wuzewu
提交者:
GitHub
8月 30, 2019
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Update usage.md
上级
bfec8688
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
1 addition
and
5 deletion
+1
-5
docs/usage.md
docs/usage.md
+1
-5
未找到文件。
docs/usage.md
浏览文件 @
95b4251e
...
...
@@ -71,9 +71,7 @@ unzip mini_pet.zip
为了方便体验,我们在configs目录下放置了Oxford-IIIT Pet所对应的配置文件
`unet_pet.yaml`
,可以通过
`--cfg`
指向该文件来设置训练配置。
我们选择两张GPU进行训练,这可以通过环境变量
`CUDA_VISIBLE_DEVICES`
来指定。
**NOTE**
: PaddleSeg会根据可用的GPU数量,将Batch内的数据平分到每张卡上,务必确保BATCH_SIZE为训练GPU数量的整数倍
我们选择GPU 0号卡进行训练,这可以通过环境变量
`CUDA_VISIBLE_DEVICES`
来指定。
```
export CUDA_VISIBLE_DEVICES=0
...
...
@@ -98,8 +96,6 @@ python pdseg/train.py --use_gpu \
> * 上述示例中,一共存在三套配置方案: PaddleSeg默认配置/unet_pet.yaml/OPTIONS,三者的优先级顺序为 OPTIONS > yaml > 默认配置。这个原则对于train.py/eval.py/vis.py/export_model.py都适用
>
> * 如果发现因为内存不足而Crash。请适当调低BATCH_SIZE。如果本机GPU内存充足,则可以调高BATCH_SIZE的大小以获得更快的训练速度
>
> * windows并不支持多卡训练
### 训练过程可视化
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录