tostring -> tobytes

This commit is contained in:
caozhou 2020-12-01 15:43:06 +08:00
parent c253d874e0
commit ad45e6bf14
1 changed files with 1 additions and 1 deletions

View File

@ -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())