mlfoundations / task_vectors

Editing Models with Task Arithmetic
433 stars 36 forks source link

Value for args.data_location #6

Open adtygan opened 1 year ago

adtygan commented 1 year ago

It's unclear to me what should I assign as the value for args.data_location. The README tells args.data_location = '/path/to/data' but I'm not sure which folder that means.

gabrielilharco commented 1 year ago

Hey @adtygan. This should be the root directory where the datasets are stored. E.g. if you have a structure like the one below, you should set args.data_location=/home/adtygan/datasets.

/home/adtygan/
      datasets/
            imagenet/
            mnist/
            ....