Update cnnTrain.m