remember to set current image dimensions of conv and pool ops.
added pickle support