Add explicit batch size parameter for FID computation

[QB3]

What does this PR do?

Add explicit batch size parameter for FID computation

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?

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 35.92%. Comparing base (fa13e7c) to head (8074f56).

Additional details and impacted files

@@           Coverage Diff           @@
##             main     #110   +/-   ##
=======================================
  Coverage   35.92%   35.92%           
=======================================
  Files          67       67           
  Lines        7399     7399           
=======================================
  Hits         2658     2658           
  Misses       4741     4741           

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

[atong01]

LGTM. Thanks!