Downloads · 30 days
3
6% of all-time downloads
dlf-ssl/ijepa-resnet50-celeba
ijepa-resnet50-celeba is a machine learning model from dlf-ssl. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for pytorch.
Downloads · 30 days
3
6% of all-time downloads
All-time downloads
51
Public
Repo size
7.2 GB
Likes
0
Public
Click a slice to open those files.
.safetensors3.6 GB · 100%
From the Hugging Face model README
Model trained using ijepa architecture.
0, 10, 100, 500, 1000
Reconstruct the model in your codebase, then load:
from huggingface_hub import hf_hub_download
from safetensors.torch import load_file
# This will automatically download just the epoch 100 file and cache it locally
ckpt_path = hf_hub_download(
repo_id="dlf-ssl/vicreg-resnet50-celeba",
filename="checkpoints/epoch_0100/model.safetensors"
)
state_dict = load_file(ckpt_path)
model.load_state_dict(state_dict, strict=False)
model.eval()