Improve saving and loading for CNN.

Now it accurately saves and loads all the CNN related parameters.
This has been tested and confirmed.