Add ability to configure model training epochs

[PawelPeczek-Roboflow]

Description

With this PR we are adding ability to configure number of epochs for training

Type of change

Please delete options that are not relevant.

• Bug fix (non-breaking change which fixes an issue)
• New feature (non-breaking change which adds functionality)
• This change requires a documentation update

How has this change been tested, please provide a testcase or example of how you tested the change?

• e2e staging

Any specific deployment considerations

For example, documentation changes, usability, usage/costs, secrets, etc.

Docs

• Docs updated? What were the changes:

[CLAassistant]

Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you all sign our Contributor License Agreement before we can accept your contribution.
1 out of 2 committers have signed the CLA.

✅ PawelPeczek-Roboflow
❌ pre-commit-ci[bot]
_{You have signed the CLA already but the status is still pending? Let us recheck it.}