Baseline TNN. Balid PPL ~ 211 with state_size=40