diff --git a/example/resnet50_imagenet2012_THOR/train.py b/example/resnet50_imagenet2012_THOR/train.py index 15710bc66be..881f3cf5983 100644 --- a/example/resnet50_imagenet2012_THOR/train.py +++ b/example/resnet50_imagenet2012_THOR/train.py @@ -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): diff --git a/mindspore/ccsrc/utils/convert_utils.cc b/mindspore/ccsrc/utils/convert_utils.cc index 6dff365504a..45c292d5454 100644 --- a/mindspore/ccsrc/utils/convert_utils.cc +++ b/mindspore/ccsrc/utils/convert_utils.cc @@ -122,7 +122,7 @@ py::object ValuePtrToPyData(const ValuePtr &value) { } else if (value->isa()) { 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; } diff --git a/mindspore/ops/operations/thor_ops.py b/mindspore/ops/operations/thor_ops.py index 9d99ae22f01..f84b5d1ffd2 100644 --- a/mindspore/ops/operations/thor_ops.py +++ b/mindspore/ops/operations/thor_ops.py @@ -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