Pass data augmentation hyperparameters as arguments

[koushik-n]

Is it possible to add these augmentation hyperparameters ( color_jitter_prob: float = 0.8, gray_scale_prob: float = 0.8, horizontal_flip_prob: float = 0.5) as arguments that can be passed to code similar to other parameters (gaussian_prob and solarization_prob)?

[vturrisi]

Hey!
This was actually supposed to be the case in our last commit, but seems like we missed this. I'll add it soon enough and also fix the default value for gray_scale_prob.

[koushik-n]

Thanks!. Also, please modify the bash file example for the custom data with these new arguments if possible!

[vturrisi]

This PR will probably address what you mentioned #164

[vturrisi]

Just merged the fixes. Should be fine now.