search

Shilin-LU / MACE

[CVPR 2024] "MACE: Mass Concept Erasure in Diffusion Models" (Official Implementation)
MIT License
272 stars 18 forks source link

Hyperparameters Used to Get Models on OneDrive #10

Open VMS-6511 opened 2 months ago

VMS-6511 commented 2 months ago

Hi,

I'm looking to reproduce the performance of the celebrity 1,5,10,100 erasure models in the paper and the ones on OneDrive. I was wondering what the hyperparameters used to train those models were? Thanks!

Best, Vinith

Shilin-LU commented 2 months ago

Hi, the hyperparameters for celebrity 1,5,10,100 are included in configs/celebrity, and you also can find them in Table 10, preserve_weight (in config) equals lambda_1 * lambda_3 (in Table 10).