Add training script with data module setup, model initialization, logger, callbacks, and trainer configuration

9f31860

bhimrazy commited on Mar 24