Remove confusing warnings during loading no-LN model on cpu

1 files changed, 19 insertions, 17 deletions

diff --git a/src/cpu/npz_converter.h b/src/cpu/npz_converter.h
index 6bd2f692..7488ead7 100644
--- a/src/cpu/npz_converter.h
+++ b/src/cpu/npz_converter.h
@@ -12,55 +12,55 @@ class NpzConverter {
       public:
         NpyMatrixWrapper(const cnpy::NpyArray& npy)
         : npy_(npy) {}
-        
+
         size_t size() const {
           return size1() * size2();
         }
-        
+
         float* data() const {
           return (float*)npy_.data;
         }
-        
+
         float operator()(size_t i, size_t j) const {
           return ((float*)npy_.data)[i * size2() + j];
         }
-        
+
         size_t size1() const {
           return npy_.shape[0];
         }
-        
+
         size_t size2() const {
           if(npy_.shape.size() == 1)
             return 1;
           else
-            return npy_.shape[1];        
+            return npy_.shape[1];
         }
-        
+
       private:
         const cnpy::NpyArray& npy_;
     };
-  
+
   public:
-    
+
     typedef blaze::CustomMatrix<float, blaze::unaligned,
       blaze::unpadded, blaze::rowMajor> BlazeWrapper;
-        
-    
+
+
     NpzConverter(const std::string& file)
       : model_(cnpy::npz_load(file)),
         destructed_(false) {
       }
-    
+
     ~NpzConverter() {
       if(!destructed_)
         model_.destruct();
     }
-    
+
     void Destruct() {
       model_.destruct();
       destructed_ = true;
     }
-    
+
     mblas::Matrix operator[](const std::string& key) const {
       BlazeWrapper matrix;
       auto it = model_.find(key);
@@ -69,14 +69,16 @@ class NpzConverter {
         matrix = BlazeWrapper(np.data(), np.size1(), np.size2());
       }
       else {
-        std::cerr << "Missing " << key << std::endl; 
+        if (key.find("gamma") == std::string::npos) {
+          std::cerr << "Missing " << key << std::endl;
+        }
       }
 
       mblas::Matrix ret;
       ret = matrix;
       return std::move(ret);
     }
-  
+
     mblas::Matrix operator()(const std::string& key,
                                    bool transpose) const {
       BlazeWrapper matrix;
@@ -91,7 +93,7 @@ class NpzConverter {
       ret = matrix2;
       return std::move(ret);
     }
-  
+
   private:
     cnpy::npz_t model_;
     bool destructed_;