Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
f40df602
MegEngine
项目概览
MegEngine 天元
/
MegEngine
9 个月 前同步成功
通知
392
Star
4702
Fork
582
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
MegEngine
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
提交
f40df602
编写于
7月 09, 2021
作者:
M
Megvii Engine Team
提交者:
huangxinda
7月 19, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
docs(mge): refactor docs to remove warnings
GitOrigin-RevId: efefc2a4a21027600d1ac3156c648bb987d039e4
上级
1040b778
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
27 addition
and
24 deletion
+27
-24
imperative/python/megengine/functional/debug_param.py
imperative/python/megengine/functional/debug_param.py
+3
-3
imperative/python/megengine/module/conv.py
imperative/python/megengine/module/conv.py
+23
-20
imperative/python/megengine/module/pooling.py
imperative/python/megengine/module/pooling.py
+1
-1
未找到文件。
imperative/python/megengine/functional/debug_param.py
浏览文件 @
f40df602
...
...
@@ -24,7 +24,7 @@ if os.getenv("MEGENGINE_CONV_EXECUTION_STRATEGY") != None:
def
get_execution_strategy
()
->
Strategy
:
"""
Returns the execution strategy of :class:`~
.Conv2d` and :func:'~.matmul'
Returns the execution strategy of :class:`~
module..Conv2d` and :func:`~.matmul`
See :func:`~.set_execution_strategy` for possible return values
"""
...
...
@@ -33,9 +33,9 @@ def get_execution_strategy() -> Strategy:
def
set_execution_strategy
(
option
):
"""
Sets the execution strategy of :class:`~
.Conv2d` and :func:'~.matmul'
Sets the execution strategy of :class:`~
module.Conv2d` and :func:`~.matmul`
:param option: Decides how :class:`~
.Conv2d`and :func:'~.matmul'
algorithms are chosen.
:param option: Decides how :class:`~
module.Conv2d`and :func:`~.matmul`
algorithms are chosen.
Available value Strategy
* HEURISTIC uses heuristic to choose the fastest algorithm.
* PROFILE runs possible algorithms on real device to find the best one.
...
...
imperative/python/megengine/module/conv.py
浏览文件 @
f40df602
...
...
@@ -144,9 +144,9 @@ class Conv1d(_ConvNd):
.. note::
:attr:`weight` usually has shape ``(out_channels, in_channels, kernel_size)``
,
if groups is not 1, shape will be ``(groups, out_channels // groups, in_channels // groups, kernel_size)``
:attr:`bias
` usually has shape ``(1, out_channels, 1)``
* ``weight`` usually has shape ``(out_channels, in_channels, kernel_size)``
,
if groups is not 1, shape will be ``(groups, out_channels // groups, in_channels // groups, kernel_size)``
* ``bias`
` usually has shape ``(1, out_channels, 1)``
Examples:
...
...
@@ -309,9 +309,9 @@ class Conv2d(_ConvNd):
.. note::
:attr:`weight` usually has shape ``(out_channels, in_channels, height, width)``
,
if groups is not 1, shape will be ``(groups, out_channels // groups, in_channels // groups, height, width)``
:attr:`bias
` usually has shape ``(1, out_channels, *1)``
* ``weight`` usually has shape ``(out_channels, in_channels, height, width)``
,
if groups is not 1, shape will be ``(groups, out_channels // groups, in_channels // groups, height, width)``
* ``bias`
` usually has shape ``(1, out_channels, *1)``
Examples:
...
...
@@ -455,9 +455,9 @@ class Conv3d(_ConvNd):
.. note::
:attr:`weight` usually has shape ``(out_channels, in_channels, depth, height, width)``
,
if groups is not 1, shape will be ``(groups, out_channels // groups, in_channels // groups, depth, height, width)``
:attr:`bias
` usually has shape ``(1, out_channels, *1)``
* ``weight`` usually has shape ``(out_channels, in_channels, depth, height, width)``
,
if groups is not 1, shape will be ``(groups, out_channels // groups, in_channels // groups, depth, height, width)``
* ``bias`
` usually has shape ``(1, out_channels, *1)``
Examples:
...
...
@@ -587,9 +587,10 @@ class ConvTranspose2d(_ConvNd):
.. note::
:attr:`weight` usually has shape ``(in_channels, out_channels, height, width)``,
if groups is not 1, shape will be ``(groups, in_channels // groups, out_channels // groups, height, width)``
:attr:`bias` usually has shape ``(1, out_channels, *1)``
* ``weight`` usually has shape ``(in_channels, out_channels, height, width)`` ,
if groups is not 1, shape will be ``(groups, in_channels // groups, out_channels // groups, height, width)``
* ``bias`` usually has shape ``(1, out_channels, *1)``
"""
def
__init__
(
...
...
@@ -685,9 +686,10 @@ class LocalConv2d(Conv2d):
.. note::
:attr:`weight` usually has shape ``(out_height, out_width, in_channels, height, width, in_channels)``,
if groups is not 1, shape will be ``(groups, out_height, out_width, in_channels // groups, height, width, out_channels // groups)``
:attr:`bias` usually has shape ``(1, out_channels, *1)``
* ``weight`` usually has shape ``(out_height, out_width, in_channels, height, width, in_channels)`` ,
if groups is not 1, shape will be ``(groups, out_height, out_width, in_channels // groups, height, width, out_channels // groups)``
* ``bias`` usually has shape ``(1, out_channels, *1)``
"""
def
__init__
(
...
...
@@ -788,9 +790,9 @@ class DeformableConv2d(_ConvNd):
.. note::
:attr:`weight` usually has shape ``(out_channels, in_channels, height, width)``
,
if groups is not 1, shape will be ``(groups, out_channels // groups, in_channels // groups, height, width)``
:attr:`bias
` usually has shape ``(1, out_channels, *1)``
* ``weight`` usually has shape ``(out_channels, in_channels, height, width)``
,
if groups is not 1, shape will be ``(groups, out_channels // groups, in_channels // groups, height, width)``
* ``bias`
` usually has shape ``(1, out_channels, *1)``
"""
...
...
@@ -898,8 +900,9 @@ class ConvTranspose3d(_ConvNd):
.. note::
:attr:`weight` usually has shape ``(in_channels, out_channels, depth, height, width)``.
:attr:`bias` usually has shape ``(1, out_channels, *1)``
* ``weight`` usually has shape ``(in_channels, out_channels, depth, height, width)`` .
* ``bias`` usually has shape ``(1, out_channels, *1)``
"""
def
__init__
(
...
...
imperative/python/megengine/module/pooling.py
浏览文件 @
f40df602
...
...
@@ -104,7 +104,7 @@ class AvgPool2d(_PoolNd):
:param stride: the stride of the window. Default value is kernel_size。
:param padding: implicit zero padding to be added on both sides.
:param mode: whether to count padding values. "average" mode will do counting and
"average_count_exclude_padding" mode won't do counting.
"average_count_exclude_padding" mode won't do counting.
Default: "average_count_exclude_padding"
Examples:
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录