提交 24bf33de 编写于 作者: M MRXLT

bug fix

上级 5c54e5c2
...@@ -114,7 +114,7 @@ int dump_parameter(const char *input_file, const char *output_file) { ...@@ -114,7 +114,7 @@ int dump_parameter(const char *input_file, const char *output_file) {
// << std::endl; // << std::endl;
memcpy(value_buf, tensor_buf + offset, value_buf_len); memcpy(value_buf, tensor_buf + offset, value_buf_len);
seq_file_writer.write( seq_file_writer.write(
static_cast<int>(&i), sizeof(i), value_buf, value_buf_len); std::to_string(i).c_str(), sizeof(i), value_buf, value_buf_len);
offset += value_buf_len; offset += value_buf_len;
} }
return 0; return 0;
...@@ -265,7 +265,7 @@ int compress_parameter_parallel(const char *file1, ...@@ -265,7 +265,7 @@ int compress_parameter_parallel(const char *file1,
SeqFileWriter seq_file_writer(file2); SeqFileWriter seq_file_writer(file2);
for (int64_t i = 0; i < dict_size; i++) { for (int64_t i = 0; i < dict_size; i++) {
seq_file_writer.write( seq_file_writer.write(
static_cast<char *>(&i), sizeof(i), result[i], per_line_size); std::to_string(i).c_str(), sizeof(i), result[i], per_line_size);
} }
return 0; return 0;
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册