Browse Source

tostring -> tobytes

tags/v1.1.0
caozhou 5 years ago
parent
commit
ad45e6bf14
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/train/serialization.py

+ 1
- 1
mindspore/train/serialization.py View File

@@ -132,7 +132,7 @@ def _exec_save(ckpt_file_name, data_list):
param_tensor = param_value.tensor param_tensor = param_value.tensor
param_tensor.dims.extend(value[0]) param_tensor.dims.extend(value[0])
param_tensor.tensor_type = value[1] 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()) f.write(checkpoint_list.SerializeToString())




Loading…
Cancel
Save