NNLM.model has attribute index2word to be saved by Train.main_loop