After adapting the repo to do more complex training, the job times out after 60 minutes. 