..

1 files changed, 9 insertions, 4 deletions

diff --git a/nerv/examples/lmptb/lm_sampler.lua b/nerv/examples/lmptb/lm_sampler.lua
index d194af9..9d31f17 100644
--- a/nerv/examples/lmptb/lm_sampler.lua
+++ b/nerv/examples/lmptb/lm_sampler.lua
@@ -8,10 +8,12 @@ function LMSampler:__init(global_conf)
     self.vocab = self.gconf.vocab
     self.sen_end_token = self.vocab.sen_end_token
     self.sen_end_id = self.vocab:get_word_str(self.sen_end_token).id 
+
+    self.loaded = false
+
 end
 
-function LMSampler:load_dagL(dagL)
-   
+function LMSampler:load_dagL(dagL)   
     nerv.printf("%s loading dagL\n", self.log_pre)
 
     self.dagL = dagL
@@ -40,9 +42,11 @@ function LMSampler:load_dagL(dagL)
         self.store[i][1].p = 0
     end
     self.repo = {}
+
+    self.loaded = true
 end
 
-function LMSampler:sample_to_store(ssout)
+function LMSampler:sample_to_store(ssout) --private
     for i = 1, self.batch_size do
         local ran = math.random()
         local id = 1
@@ -88,8 +92,9 @@ function LMSampler:sample_to_store(ssout)
     end
 end
 
---Returns: LMResult
 function LMSampler:lm_sample_rnn_dagL(sample_num, p_conf)
+    assert(self.loaded == true)
+
     local dagL = self.dagL
     local inputs = self.dagL_inputs
     local outputs = self.dagL_outputs