search

OPTML-Group / UnlearnCanvas

UnlearnCanvas: A Stylized Image Dataaset to Benchmark Machine Unlearning for Diffusion Models by Yihua Zhang, Chongyu Fan, Yimeng Zhang, Yuguang Yao, Jinghan Jia, Jiancheng Liu, Gaoyuan Zhang, Gaowen Liu, Ramana Kompella, Xiaoming Liu, Sijia Liu
https://unlearn-canvas.netlify.app
48 stars 2 forks source link

Difference between the ckpts? #6

Closed RichardSunnyMeng closed 3 months ago

RichardSunnyMeng commented 3 months ago

Hi, authors. I want to use the style classifier but don't know the differences between style50_cls.pth and style50.pth. What is the difference between them? And I find that there are 51 style classes in the codes but the file names indicate that there are 50 classes?

NormalUhr commented 3 months ago

Hi, thanks for your interest in our work. The style50_cls.pth is the checkpoint for vit to do object-wise classification, so it has a classification head with only 20 classes. The style50.pth aims to do style classification. Its classification head has 51 classes because there are 50 artistic styles + 1 photo style (no artistic style). Hope this answers your questions.

RichardSunnyMeng commented 3 months ago

Thanks for your careful comment!