Created by: jetfuel

This updates the train.py file of Image Classification. The update for Markdown will follow