* Update train.py

As see in #6463, modification on train in evolve process to allow custom save directory.

* fix val

* PEP8

whitespace around operator

* Cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>