CASIA-IVA-Lab/AnomalyGPT

有监督训练时,除了将预训练模型更换之外,还需要更改其他地方吗?

William9Baker opened this issue · 1 comments

有监督训练时,除了将预训练模型更换之外,还需要更改其他地方吗?比如是否需要将数据集加载的函数由load_mvtec_dataset()改为load_supervised_dataset_with_cn()?

需要的,你可以参考我们提供的有监督训练代码 train_all_supervised_cn.py 对比查看修改了哪些地方