search

TRAIS-Lab / dattri

`dattri` is a PyTorch library for developing, benchmarking, and deploying efficient data attribution algorithms.
https://trais-lab.github.io/dattri/
24 stars 8 forks source link

[dattri.benchmark] Fix output format of maestro dataset creation #69

Closed tingwl0122 closed 3 months ago

tingwl0122 commented 3 months ago

Description

This PR fix the output format of the create_maestro_datasets function.

1. Motivation and Context

Originally, the dattri/benchmark/maestro/data/create_maestro_datasets function will directly output three dataloaders. But I found that the current dattri/scripts/dattri_retrain.py actually will assume this function to output torch dataset objects (will construct data loaders in that script). Thus make this change.

2. Summary of the change

TheaperDeng commented 3 months ago

Please merge this PR if there is nothing to change.