p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix relative import path for RDR st

This commit is contained in:
liangyongxiong 2021-12-02 19:12:58 +08:00
parent 32a38810ed
commit 94298e9b90
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):