Fix Memory Leak

[hukaixuan19970627]

有个动态分配内存的变量没被释放，只有在训练程序终止的时候才会得到释放。 训练时间过长或数据集较大时会引发内存不足的问题。