VAND-APRIL-GAN icon indicating copy to clipboard operation
VAND-APRIL-GAN copied to clipboard

Published 20 hours ago • verified publisher icon ByChelsea

mvtec epoch

Open Wangjinhong1998 opened this issue 1 year ago • 1 comments

Why did you set the epoch to 3 when training the MVTec dataset, but to 15 when training the Visa dataset? I noticed that the loss on MVTec was still decreasing after the third epoch

Wangjinhong1998 avatar Jan 19 '24 09:01 Wangjinhong1998

Sorry for the late reply.

We hope that the added linear layers learn a general mapping, rather than one that is specific to a particular dataset. Therefore, when training reaches a certain point, continued reduction in loss may lead to decreased generalization, which is not desirable.

It is worth noting that it's indeed challenging to determine the best time to stop training, which is a problem faced by all fine-tuning zero-shot methods currently (AnomalyCLIP, CLIP-AD). I suggest using three different datasets for train, valid, and test respectively, as a intuitive solution. I also look forward to future works addressing this issue.

ByChelsea avatar Jan 31 '24 08:01 ByChelsea