[AI] updates to LeNet4