tostring -> tobytes

4 years ago · ad45e6bf14
parent c253d874e0
commit ad45e6bf14
1 changed files with 1 additions and 1 deletions
--- a/mindspore/train/serialization.py
+++ b/mindspore/train/serialization.py
@ -132,7 +132,7 @@ def _exec_save(ckpt_file_name, data_list):
                        param_tensor = param_value.tensor
                        param_tensor.dims.extend(value[0])
                        param_tensor.tensor_type = value[1]
-                        param_tensor.tensor_content = param_slice.tostring()
+                        param_tensor.tensor_content = param_slice.tobytes()
                        f.write(checkpoint_list.SerializeToString())