Removed warnings and testing lines.
CNN is fully trainable, but no predict function.