Setup TensorBoard with logs

[arun-gupta]

• Need to change model to dump training logs to disk
• These need to be stored in S3 bucket
• Then follow the steps at https://github.com/aws-samples/machine-learning-using-k8s/blob/master/docs/tensorboard.md to show TensorBoard visualization

[arun-gupta]

@Jeffwan will look at this