aboutsummaryrefslogblamecommitdiff
path: root/nerv/examples/lmptb/lm_trainer.lua
blob: 8a9744b2909ec6b6f7806a6c71f4f9e5394f1b3f (plain) (tree)
1
2
3
4
5
6
7
8
9



                          
                    



                                              
                            
 






                                                      
                   




                                                                              
                     


                                                         
                                
                                                                                                                  
                                
                                                                                                                                                  
           
                                                                                                                                           
                                           
                                            
                                
        

                                           
       
 
                                                                                             
                        
 

                                                                


                                                                                                
        
                             
                                    


                                                                          
                                                                                                   

                                                                   
                

                                                            
                      
                                                  



                                                  
    
                                
                                          
                                    
                                                                                   



                                                        
                                                  












                                                                                                                       
                                


                                         

                                                 
                           
                            
                                   
                             
                                                   
                                       
                                                                              

                                                                                                      
                                                                   



                                                                   


                           










                                                                                                                                                   

                                             


                                                                                                             


               


                                                            
                                                            

                   
                                                 
                                                               

                                                                                                                
                                                             
                                                                                                     

                                     








                                            




                                                                               


                                                                          



                 







                                                                                


                                                         



                                                                                                                                 
                                                                                                       
                                        
                                           





                                            
                                                                                             








                                                                                                  
                                    




                                                                                                   

                                                                     








                                                            










































                                                                                                                       


                                                                                                                





                                                                                      

                                                                                                            































                                                                                                                  
 
require 'lmptb.lmvocab'
require 'lmptb.lmfeeder'
require 'lmptb.lmutil'
require 'lmptb.layer.init'
--require 'tnn.init'
require 'lmptb.lmseqreader'

local LMTrainer = nerv.class('nerv.LMTrainer')

--local printf = nerv.printf

--The bias param update in nerv don't have wcost added
function nerv.BiasParam:update_by_gradient(gradient) 
    local gconf = self.gconf
    local l2 = 1 - gconf.lrate * gconf.wcost
    self:_update_by_gradient