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_retrain.py` need to set `start_id` from `partition` argument #89

Closed tingwl0122 closed 1 month ago

tingwl0122 commented 3 months ago

Add one more line at the bottom:

    if args.mode == "lds":
        kwargs = {}
        kwargs["num_subsets"] = int(args.partition[1]) - int(args.partition[0])
        kwargs["total_num_subsets"] = int(args.partition[2])
        kwargs["start_id"] = int(args.partition[0])