do you have your Task.init call inside the "train.py" script ? (and if you do, what are you getting in the Execution tab of the task) ?