search

zqhang / AnomalyCLIP

Official implementation for AnomalyCLIP (ICLR 2024)
MIT License
269 stars 26 forks source link

训练集所选辅助数据训练疑问 #39

Open 2402227288 opened 3 months ago

2402227288 commented 3 months ago

作者您好，对于train.sh，CUDA_VISIBLE_DEVICES=${device} python train.py --dataset mvtec --train_data_path，dataset和train_data_path分别代表什么呢，因为您文章指出mvtec用的visa训练，visa用的mvtec训练，我看您代码这两部分训练集前后都一致

zqhang commented 2 months ago

你好，dataset是数据集的名字，path是相应数据集存储路径。train.sh只是用来训练模型，test.sh是进行zero-shot测试。