add ae initialization to cnn