initialize_weights_with, discount_coeff for augmentation
spatial augmentation of multi-channel images
bunch of small changes