Skip to content

Releases: DeepLearnPhysics/PoLAr-MAE

Model weights

06 Feb 07:26
Compare
Choose a tag to compare
  • polarmae_pretrain.ckpt: PoLAr-MAE pre-trained for 500k steps
  • mae_pretrain.ckpt Point-MAE pre-trained for 500k steps.
  • mae_peft_segseg_10k.ckpt: Point-MAE PEFT for 100k steps on 10k events using mae_pretrain.ckpt as starting point.
  • mae_fft_segseg_10k.ckpt: Point-MAE FFT for 100k steps on 10k events using mae_pretrain.ckpt as starting point.
  • polarmae_peft_segseg_10k.ckpt: PoLAr-MAE PEFT for 100k steps on 10k events using polarmae_pretrain.ckpt as starting point.
  • polarmae_fft_segseg_10k.ckpt: PoLAr-MAE FFT for 100k steps on 10k events using polarmae_pretrain.ckpt as starting point.