Update train.py#4

elleobrien wants to merge 3 commits into

masteriterative/cml_dvc_case:masterfrom

experimentiterative/cml_dvc_case:experimentCopy head branch name to clipboard