fixed bug in deep learning