cifar10: try unblock training

[Borda]

Before submitting

• Was this discussed/approved via a Github issue? (no need for typos and docs improvements)
• Did you make sure to update the docs?
• Did you write any new necessary tests?

What does this PR do?

the notebooks takes/hangs for 3 hours which is above the expectation limits

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?

Make sure you had fun coding 🙃

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 70%. Comparing base (7b418e8) to head (44ada63).
Report is 1 commits behind head on main.

Additional details and impacted files

@@         Coverage Diff         @@
##           main   #335   +/-   ##
===================================
- Coverage    70%    70%   -0%     
===================================
  Files         2      2           
  Lines       406    405    -1     
===================================
- Hits        285    284    -1     
  Misses      121    121