update image_classification_with_densenet/image_cnn_train/image_classification/training.py

[yuyue9284]

PR into Azure/azureml-examples

Checklist

I have:

• [x] read and followed the contributing guidelines

Changes

Update training scripts so that the example can runs on AmlArc compute target with common runtime.