* add datamodule with crop logic * remove print and fix batch_size * typo * Use Kornia augmentations * Style * Ignore warning Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>