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

!1734 deal with resnet50_THOR train print many ERROR&WARNING log and produce many ir file 

Merge pull request !1734 from zongha/r0.3
This commit is contained in:
mindspore-ci-bot 2020-06-22 15:56:13 +08:00 committed by Gitee
parent cf7c60a5ed 491ba51b8b
commit 4bbd4414c4
3 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
example/resnet50_imagenet2012_THOR/train.py
View File

@ -46,7 +46,7 @@ parser.add_argument('--dataset_path', type=str, default=None, help='Dataset path
args_opt = parser.parse_args()
device_id = int(os.getenv('DEVICE_ID'))
context.set_context(mode=context.GRAPH_MODE, device_target="Ascend", save_graphs=True, device_id=device_id)
context.set_context(mode=context.GRAPH_MODE, device_target="Ascend", save_graphs=False, device_id=device_id)
def get_model_lr(global_step, lr_init, decay, total_epochs, steps_per_epoch):

2
mindspore/ccsrc/utils/convert_utils.cc
View File

@ -121,7 +121,7 @@ py::object ValuePtrToPyData(const ValuePtr &value) {
} else if (value->isa<None>()) {
ret = py::none();
} else {
MS_LOG(EXCEPTION) << "Unsupported convert value: " << value->ToString() << " to a PyData.";
MS_LOG(INFO) << "Unsupported convert value: " << value->ToString() << " to a PyData.";
}
return ret;
}

6
mindspore/ops/operations/thor_ops.py
View File

@ -96,6 +96,12 @@ class CusCholeskyTrsm(PrimitiveWithInfer):
"""init CusCholeskyTrsm"""
self.init_prim_io_names(inputs=['x1'], outputs=['y'])
from mindspore.ops._op_impl._custom_op.cholesky_trsm_impl import CusCholeskyTrsm
def get_bprop(self):
def bprop(x, out, dout):
return (C.zeros_like(x),)
return bprop
def infer_shape(self, data1_shape):
ll = []
m, _ = data1_shape