Refactor ImageNet workloads to train out-of-the-box

[svrama]

The ImageNet workloads require some cleanup to train out-of-the-box with the latest version of TensorFlow after a clean Fathom install.

Currently, this needs some plug-and-play and editing of hyperparameters.