mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!27145 fix relative import path for RDR st 

Merge pull request !27145 from liangyongxiong/fix
This commit is contained in:
wmzheng2020 2021-12-02 14:19:46 +00:00 committed by Gitee
parent eb25c473d0 94298e9b90
commit 96979f05cc
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/st/networks/test_network_main.py
View File

@ -21,9 +21,6 @@ Usage:
import argparse
import numpy as np
from models.alexnet import AlexNet
from models.lenet import LeNet
from models.resnetv1_5 import resnet50
import mindspore.context as context
import mindspore.nn as nn
@ -31,7 +28,9 @@ from mindspore import Tensor
from mindspore.nn import TrainOneStepCell, WithLossCell
from mindspore.nn.optim import Momentum
context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
from .models.alexnet import AlexNet
from .models.lenet import LeNet
from .models.resnetv1_5 import resnet50
def train(net, data, label):