search

mlfoundations / open_clip

An open source implementation of CLIP.
Other
9.93k stars 959 forks source link

key error #789

Closed ntz2000 closed 5 months ago

ntz2000 commented 9 months ago

use "python -m training.main --save-frequency 1 --zeroshot-frequency 1 --report-to tensorboard --train-data="test.csv" --csv-img-key filepath --csv-caption-key title --warmup 10000 --batch-size=128 --lr=1e-3 --wd=0.1 --epochs=30 --workers=1 --model RN50"

And the test.csv file has the same col but :

raise KeyError(key) from err KeyError: 'filepath'

StefanDenn3r commented 8 months ago

Hi @ntz2000, It seems like the issue might be related to the CSV separator used in your command. The default separator for the --csv-separatoroption is set to '\t' (tab-separated). If your test.csv file uses a different separator, such as a comma or a semicolon, you need to specify this in your command.

Try adding the --csv-separator flag with the appropriate separator. For example, if your CSV file is comma-separated, modify your command to include --csv-separator ','. Make sure to match the separator with the actual format of your test.csv file.