srl create model bug

XiaoboYuan · May 8, 2014 · c4bfbe9 · c4bfbe9
1 parent b06be2d
commit c4bfbe9
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/srl/DepSRL.cpp b/src/srl/DepSRL.cpp
@@ -21,10 +21,16 @@ int DepSRL::LoadResource(const string &ConfigDir)
     m_selectFeats = ConfigDir + "/srl.cfg";
     // load srl and prg model
     m_srlModel = new maxent::ME_Model;
-    m_srlModel->load(ConfigDir + "/srl.model");
+    bool tag = m_srlModel->load(ConfigDir + "/srl.model");
+    if(!tag) {
+      return 0;
+    }
 
     m_prgModel = new maxent::ME_Model;
-    m_prgModel->load(ConfigDir + "/prg.model");
+    tag = m_prgModel->load(ConfigDir + "/prg.model");
+    if(!tag) {
+      return 0;
+    }
 
     m_resourceLoaded = true;