memory convolv layer train