Fixed the deprecations warning, was a configuration issue. First training run completed successfully.