set save graphs False and add bprop for op cholesky trsm

Change thor log level
pull/1722/head
z00478463 5 years ago
parent 0920660ecc
commit f8dda8d3b1

@ -46,7 +46,7 @@ parser.add_argument('--dataset_path', type=str, default=None, help='Dataset path
args_opt = parser.parse_args() args_opt = parser.parse_args()
device_id = int(os.getenv('DEVICE_ID')) 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): def get_model_lr(global_step, lr_init, decay, total_epochs, steps_per_epoch):

@ -122,7 +122,7 @@ py::object ValuePtrToPyData(const ValuePtr &value) {
} else if (value->isa<None>()) { } else if (value->isa<None>()) {
ret = py::none(); ret = py::none();
} else { } 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; return ret;
} }

@ -96,6 +96,12 @@ class CusCholeskyTrsm(PrimitiveWithInfer):
"""init CusCholeskyTrsm""" """init CusCholeskyTrsm"""
self.init_prim_io_names(inputs=['x1'], outputs=['y']) self.init_prim_io_names(inputs=['x1'], outputs=['y'])
from mindspore.ops._op_impl._custom_op.cholesky_trsm_impl import CusCholeskyTrsm 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): def infer_shape(self, data1_shape):
ll = [] ll = []
m, _ = data1_shape m, _ = data1_shape

Loading…
Cancel
Save