1 files changed, 1 insertions, 0 deletions
diff --git a/train.cpp b/train.cpp
index 04e80ce..83d8cdc 100644
--- a/train.cpp
+++ b/train.cpp
@@ -121,6 +121,7 @@ void getmask(string fname,vector<double> &msk)
     int fsize;
     fin.open(fname.c_str());
     fin>>fsize;
+    msk.resize(fsize);
     for (int i=0;i<fsize;++i)
         fin>>msk[i];
     fin.close();