Skip to content
Snippets Groups Projects
Unverified Commit eaf92620 authored by i-robot's avatar i-robot Committed by Gitee
Browse files

!2196 resnet update deprecated API

Merge pull request !2196 from wangzeyangyi/resnet
parents 1d092a03 58307a2c
No related branches found
No related tags found
No related merge requests found
......@@ -55,7 +55,7 @@ def infer_net():
target = config.device_target
# init context
ms.set_context(mode=context.GRAPH_MODE, device_target=target, save_graphs=False)
ms.set_context(mode=ms.GRAPH_MODE, device_target=target, save_graphs=False)
if target == "Ascend":
device_id = int(os.getenv('DEVICE_ID'))
ms.set_context(device_id=device_id)
......
......@@ -15,6 +15,7 @@
"""ResNet."""
import numpy as np
from scipy.stats import truncnorm
import mindspore as ms
import mindspore.nn as nn
from mindspore.ops import operations as P
from mindspore.common.tensor import Tensor
......
......@@ -144,7 +144,7 @@ def set_parameter():
ms.set_context(mode=ms.PYNATIVE_MODE, device_target=target, save_graphs=False)
if config.parameter_server:
context.set_ps_context(enable_ps=True)
ms.set_ps_context(enable_ps=True)
if config.run_distribute:
if target == "Ascend":
device_id = int(os.getenv('DEVICE_ID'))
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment