Update train.py#11

elleobrien wants to merge 2 commits intomasteriterative/cml_dvc_case:masterfrom

experiment-1iterative/cml_dvc_case:experiment-1Copy head branch name to clipboard