Move batching to samplers (maybe)

aai-institute / pyDVL

pyDVL is a library of stable implementations of algorithms for data valuation and influence function computation

https://pydvl.org

GNU Lesser General Public License v3.0

89 stars 9 forks source link

Move batching to samplers (maybe) #433

Open mdbenito opened 9 months ago

mdbenito commented 9 months ago

If we keep the batching hack, I think it belongs in the samplers. A simple way to do it is to wrap them in a generic batched decorator like this one.