[fix] benchmark shows 0.00MB consumed

[deciding]

What does this PR do?

Fixes bugs: when run pytest, the memory consumption printed is 0.00MB.
This is because class creation (also weight creation) happens in benchmarks_generator. Which is outside the torch.cuda.reset_peak_memory_stats() and torch.cuda.max_memory_allocated() block.

Before submitting

• Did you have fun?
  • Make sure you had fun coding 🙃
• Did you read the contributor guideline?
• Was this discussed/approved via a Github issue? (no need for typos, doc improvements)
  • N/A
• Did you make sure to update the docs?
  • N/A
• Did you write any new necessary tests?
  • N/A
• Did you update the changelog? (if needed)
  • N/A

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.

[codecov-commenter]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 60.56%. Comparing base (b6b68e3) to head (68a8b51).

Additional details and impacted files

@@           Coverage Diff           @@
##             main    #1018   +/-   ##
=======================================
  Coverage   60.56%   60.56%           
=======================================
  Files         113      113           
  Lines        9820     9820           
=======================================
  Hits         5947     5947           
  Misses       3873     3873           

Flag	Coverage Δ
Python	60.56% <ø> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.