AMSS文件中的train_test.py文件存在问题

[pinglanjiu]

train_epoch函数中的FFN层掩码子网出现了问题。只构建好了mask，缺少将子网应用到梯度上的代码。