关于复现效果问题

跑了下这个代码，效果和论文中单layer也就是 DMF-1-nce 差不多；
![image](https://user-images.githubusercontent.com/3424112/50480089-f191d200-0a14-11e9-9b14-4f5617642258.png)

怎么修改参数去对照论文中的两层效果呢？