fix max_epochs setup in basic example

[oplatek]

tested only for the CPU version

Before submitting

• Was this discussed/approved via a Github issue? (no need for typos and docs improvements)
• Did you read the contributor guideline, Pull Request section?
• Did you make sure to update the docs?
• Did you write any new necessary tests?
• If you made a notable change (that affects users), did you update the CHANGELOG?

What does this PR do?

Fixes the issue that the max_epochs parameter is ignored.

Tested via running it locally only

python cpu_template.py --epochs 1

PR review

Anyone in the community is free to review the PR once the tests have passed.
If we didn't discuss your PR in Github issues there's a high chance it will not be merged.

Did you have fun?

🌞

[oplatek]

Also, consider removing the argparse argument

parser.add_argument('--optimizer_name', default='adam', type=str)

because it is not used in the code.

[Borda]

hey there, we have added GPU CI test, so could we kindly ask to rebase/merge master which will trigger these tests so we do not need to test it manually... Thx for your understanding 🤖

[oplatek]

@Borda I just force pushed the branch where I rebased the upstream/master (upstream == PL/PL master)

[Borda]

LGTM 🚀