pt: support list format batch size

[CaRoLZhangxy]

#3475

[codecov]

Codecov Report

Attention: Patch coverage is 73.68421% with 5 lines in your changes are missing coverage. Please review.

Project coverage is 77.72%. Comparing base (c1f22b5) to head (503c398).
Report is 3 commits behind head on devel.

Files	Patch %	Lines
deepmd/pt/utils/dataloader.py	73.68%	5 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##            devel    #3614      +/-   ##
==========================================
+ Coverage   77.71%   77.72%   +0.01%     
==========================================
  Files         433      434       +1     
  Lines       37471    37530      +59     
  Branches     1620     1620              
==========================================
+ Hits        29120    29172      +52     
- Misses       7490     7497       +7     
  Partials      861      861              

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