Change thor log leveltags/v0.5.0-beta
| @@ -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 | ||||