Added basic logger configuration for ./train.py