我该如何训练自己的数据集？

[Enuang]

感谢你做出的工作，我是一个初学者，我需要如何进行对比学习预训练，如何用在我自己的细粒度视觉分类数据集上呢？是修改main.py中的参数is_pretrain吗？

[Markin-Wang]

Hi, 感谢你的兴趣，首先你需要参考data/dataset.py文件里数据集的定义来写一个自己的数据集的类来读取，处理，返回数据集。之后需要参考在scripts文件夹里的脚本写一个自己数据集的运行脚本，特别要更改dataset这个参数为你自己定义的数据集的名字。可能还需要一些小调整的地方，希望这能帮助你在你自己数据集上运行。