Added basic logger configuration for ./train.py

Adding logger configuration for ./train.py. Without this, the warning message will not show up and, hence, can be trouble to newbies.