Downloads · 30 days
4
8% of all-time downloads
DeepLearnPhysics/PoLAr-MAE-Pretrain
PoLAr-MAE-Pretrain is a machine learning model from DeepLearnPhysics. 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 pimm.
Full PoLAr-MAE masked-autoencoder pretrained on LArTPC (PILArNet) point clouds (masked point reconstruction + energy infilling): ViT-S encoder, MAE decoder, reconstruction heads. Use the encoder as a backbone / warm-s…
Downloads · 30 days
4
8% of all-time downloads
All-time downloads
48
Public
Parameters
29.6M
119 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors119 MB · 100%
From the Hugging Face model README
Full PoLAr-MAE masked-autoencoder pretrained on LArTPC (PILArNet) point clouds (masked point reconstruction + energy infilling): ViT-S encoder, MAE decoder, reconstruction heads. Use the encoder as a backbone / warm-start.
PoLAr-MAE · arch vit_smallimport pimm
model = pimm.from_pretrained("hf://deeplearnphysics/polar-mae-pretrain")
Repackaged from the original PoLAr-MAE release checkpoints into the pimm export format. Inherits the source repo license.