fix: mindrecord seekg failed when shift raw page

5 years ago · 2aaf3a37d4
parent 95ef02af7d
commit 2aaf3a37d4
1 changed files with 1 additions and 1 deletions
--- a/mindspore/ccsrc/mindrecord/io/shard_writer.cc
+++ b/mindspore/ccsrc/mindrecord/io/shard_writer.cc
@ -90,7 +90,7 @@ MSRStatus ShardWriter::OpenDataFiles(bool append) {
      fs->close();

      // open the mindrecord file to write
-      fs->open(common::SafeCStr(file), std::ios::out | std::ios::binary);
+      fs->open(common::SafeCStr(file), std::ios::out | std::ios::in | std::ios::binary | std::ios::trunc);
      if (!fs->good()) {
        MS_LOG(ERROR) << "MindRecord file could not opened.";
        return FAILED;