TRAIS-Lab / dattri

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

[dattri.algorithm.tracin.py] tiny but misleading typo for test set #120

Closed KurisuTheAmadeus closed 5 days ago

KurisuTheAmadeus commented 1 month ago

in dattri.algorithm.tracin.py, from line 139:

                for test_batch_idx, test_batch_data_ in enumerate(
                    tqdm(
                        test_dataloader,
                        desc="calculating gradient of training set...",
                        leave=False,
                    ),
                )

the desc should be "calculating gradient of test set..." instead. This is just a small typo but it is a little bit misleading.

jiaqima commented 5 days ago

Fixed in #134