Skip to content
Snippets Groups Projects
Commit 58307a2c authored by wangzeyangyi's avatar wangzeyangyi
Browse files

updated deprecated API

parent bdf2d8bc
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