提交 5ea13a79 编写于 作者: S simson

fix bug of examples

上级 800e0fc7
...@@ -192,6 +192,7 @@ from mindspore import context, Tensor ...@@ -192,6 +192,7 @@ from mindspore import context, Tensor
from mindspore.common.api import ms_function from mindspore.common.api import ms_function
context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU") context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
conv_obj.init_parameters_data()
conv_obj = nn.Conv2d(in_channels=3, out_channels=4, kernel_size=3, stride=2, padding=0) conv_obj = nn.Conv2d(in_channels=3, out_channels=4, kernel_size=3, stride=2, padding=0)
@ms_function @ms_function
......
...@@ -192,6 +192,7 @@ from mindspore import context, Tensor ...@@ -192,6 +192,7 @@ from mindspore import context, Tensor
from mindspore.common.api import ms_function from mindspore.common.api import ms_function
context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU") context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
conv_obj.init_parameters_data()
conv_obj = nn.Conv2d(in_channels=3, out_channels=4, kernel_size=3, stride=2, padding=0) conv_obj = nn.Conv2d(in_channels=3, out_channels=4, kernel_size=3, stride=2, padding=0)
@ms_function @ms_function
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册