Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MindSpore
book
提交
bacd6196
B
book
项目概览
MindSpore
/
book
通知
3
Star
1
Fork
1
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
B
book
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
bacd6196
编写于
9月 04, 2020
作者:
M
mindspore-ci-bot
提交者:
Gitee
9月 04, 2020
浏览文件
操作
浏览文件
下载
差异文件
!20 modify lenet
Merge pull request !20 from wukesong/modify_lenet_pa
上级
d043b195
8dcaf8b1
变更
2
显示空白变更内容
内联
并排
Showing
2 changed file
with
9 addition
and
9 deletion
+9
-9
chapter03/lenet/lenet.py
chapter03/lenet/lenet.py
+8
-8
chapter03/lenet/main.py
chapter03/lenet/main.py
+1
-1
未找到文件。
chapter03/lenet/lenet.py
浏览文件 @
bacd6196
...
@@ -14,6 +14,7 @@
...
@@ -14,6 +14,7 @@
# ============================================================================
# ============================================================================
"""LeNet."""
"""LeNet."""
import
mindspore.nn
as
nn
import
mindspore.nn
as
nn
from
mindspore.common.initializer
import
Normal
class
LeNet5
(
nn
.
Cell
):
class
LeNet5
(
nn
.
Cell
):
...
@@ -22,22 +23,21 @@ class LeNet5(nn.Cell):
...
@@ -22,22 +23,21 @@ class LeNet5(nn.Cell):
Args:
Args:
num_class (int): Num classes. Default: 10.
num_class (int): Num classes. Default: 10.
channel (int): Num classe
s. Default: 1.
num_channel (int): Num channel
s. Default: 1.
Returns:
Returns:
Tensor, output tensor
Tensor, output tensor
Examples:
Examples:
>>> LeNet(num_class=10, channel=1)
>>> LeNet(num_class=10,
num_
channel=1)
"""
"""
def
__init__
(
self
,
num_class
=
10
,
channel
=
1
):
def
__init__
(
self
,
num_class
=
10
,
num_
channel
=
1
):
super
(
LeNet5
,
self
).
__init__
()
super
(
LeNet5
,
self
).
__init__
()
self
.
num_class
=
num_class
self
.
conv1
=
nn
.
Conv2d
(
num_channel
,
6
,
5
,
pad_mode
=
'valid'
)
self
.
conv1
=
nn
.
Conv2d
(
channel
,
6
,
5
,
pad_mode
=
'valid'
)
self
.
conv2
=
nn
.
Conv2d
(
6
,
16
,
5
,
pad_mode
=
'valid'
)
self
.
conv2
=
nn
.
Conv2d
(
6
,
16
,
5
,
pad_mode
=
'valid'
)
self
.
fc1
=
nn
.
Dense
(
16
*
5
*
5
,
120
)
self
.
fc1
=
nn
.
Dense
(
16
*
5
*
5
,
120
,
weight_init
=
Normal
(
0.02
)
)
self
.
fc2
=
nn
.
Dense
(
120
,
84
)
self
.
fc2
=
nn
.
Dense
(
120
,
84
,
weight_init
=
Normal
(
0.02
)
)
self
.
fc3
=
nn
.
Dense
(
84
,
self
.
num_class
)
self
.
fc3
=
nn
.
Dense
(
84
,
num_class
,
weight_init
=
Normal
(
0.02
)
)
self
.
relu
=
nn
.
ReLU
()
self
.
relu
=
nn
.
ReLU
()
self
.
max_pool2d
=
nn
.
MaxPool2d
(
kernel_size
=
2
,
stride
=
2
)
self
.
max_pool2d
=
nn
.
MaxPool2d
(
kernel_size
=
2
,
stride
=
2
)
self
.
flatten
=
nn
.
Flatten
()
self
.
flatten
=
nn
.
Flatten
()
...
...
chapter03/lenet/main.py
浏览文件 @
bacd6196
...
@@ -91,7 +91,7 @@ if __name__ == "__main__":
...
@@ -91,7 +91,7 @@ if __name__ == "__main__":
context
.
set_context
(
mode
=
context
.
GRAPH_MODE
,
device_target
=
args
.
device_target
)
context
.
set_context
(
mode
=
context
.
GRAPH_MODE
,
device_target
=
args
.
device_target
)
network
=
LeNet5
(
cfg
.
num_classes
)
network
=
LeNet5
(
cfg
.
num_classes
)
net_loss
=
nn
.
SoftmaxCrossEntropyWithLogits
(
is_grad
=
False
,
sparse
=
True
,
reduction
=
"mean"
)
net_loss
=
nn
.
SoftmaxCrossEntropyWithLogits
(
sparse
=
True
,
reduction
=
"mean"
)
repeat_size
=
1
repeat_size
=
1
net_opt
=
nn
.
Momentum
(
network
.
trainable_params
(),
cfg
.
lr
,
cfg
.
momentum
)
net_opt
=
nn
.
Momentum
(
network
.
trainable_params
(),
cfg
.
lr
,
cfg
.
momentum
)
model
=
Model
(
network
,
net_loss
,
net_opt
,
metrics
=
{
"Accuracy"
:
Accuracy
()})
model
=
Model
(
network
,
net_loss
,
net_opt
,
metrics
=
{
"Accuracy"
:
Accuracy
()})
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录