Improve comments

[ashleve]

What does this PR do?

Improves comments in train.py and eval.py.

Before submitting

• Did you make sure title is self-explanatory and the description concisely explains the PR?
• Did you make sure your PR does only one thing, instead of bundling different changes together?
• Did you list all the breaking changes introduced by this pull request?
• Did you test your PR locally with pytest command?
• Did you run pre-commit hooks with pre-commit run -a command?

Did you have fun?

Make sure you had fun coding 🙃

[codecov-commenter]

Codecov Report

Merging #429 (4b7a3d0) into main (1ad3c2e) will not change coverage.
The diff coverage is 100.00%.

@@           Coverage Diff           @@
##             main     #429   +/-   ##
=======================================
  Coverage   84.59%   84.59%           
=======================================
  Files           9        9           
  Lines         344      344           
=======================================
  Hits          291      291           
  Misses         53       53           

Impacted Files	Coverage Δ
src/eval.py	90.32% <100.00%> (ø)
src/train.py	95.83% <100.00%> (ø)

Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here.