check format before set header format

8 years ago · 4d8992c3bc
parent 940b4a9c82
commit 4d8992c3bc
1 changed files with 5 additions and 1 deletions
--- a/paddle/parameter/Parameter.h
+++ b/paddle/parameter/Parameter.h
@ -278,7 +278,11 @@ public:
  /**
   * @brief Set the format in header.
   */
-  void setHeaderFormat(int32_t fmt) { headerFormat_ = fmt; }
+  void setHeaderFormat(int32_t fmt) {
+    CHECK(isHeaderFormatSupported(fmt)) << "Unsupported format version: "
+                                        << fmt;
+    headerFormat_ = fmt;
+  }

  /**
   * @brief  Parameter Update Hook.