Additional features and code improvements

• Override config parameters from command-line in prediction
• Move tensorboard files to separate subfolders in CHECKPOINT_DIR
• Allow other optimizers to be used apart from Adam