Made changes in the training script

Re training will finish this time for sure

Print statements for logging

Change any output errors

Change the config file accordingly