flwrlabs/celeba
Viewer β’ Updated β’ 203k β’ 5.23k β’ 18
This repository provides pretrained models used in our study on inversion attacks against DataMix.
The models are based on a ResNet-18 architecture and are trained on a mixed version of the CelebA dataset. They are specifically designed for evaluating privacy vulnerabilities and recovering private mixing coefficients via inversion attacks.
For full details about the methodology, training setup, and data preprocessing, please refer to the main project repository:
π https://github.com/hehteram/Breaking-DataMix
from huggingface_hub import hf_hub_download
import torch
path = hf_hub_download(
repo_id="hehteram/Breaking-DataMix",
filename="models/model.pth"
)
model.load_state_dict(torch.load(path))
Please refer to the main GitHub repository for licensing and usage terms.
Base model
microsoft/resnet-18