About early stopping during training

SunnyHaze / IML-ViT

Official repository of paper “IML-ViT: Benchmarking Image manipulation localization by Vision Transformer”

MIT License

184 stars 23 forks source link

About early stopping during training #15

Closed Shimjoonkyo closed 4 months ago

Shimjoonkyo commented 5 months ago

Hello, i have a question about training settings in Implementation Details, which say "Evaluate the performance on the F1 score for CASIAv1, and stop training when there is no improvement for 15 epochs."

Is it right that the you choose the best model based on the TEST dataset results (CASIAv1) ?

SunnyHaze commented 5 months ago

Thanks for your attention!

Yes. We select the best model from the best pixel-level F1 result obtained from CASIAv1.

SunnyHaze commented 4 months ago

Since we haven't heard back from the original poster for some time, I'll close this issue. Please feel free to reopen it if you need to. Thanks!