UniFault-small

PHMFD pretrained feature extractor exported from the local main-branch pretraining log. This is the PHMFD implementation of unifault, not an upstream official checkpoint.

  • Training seed: 42; checkpoint epoch: 5.
  • Feature dimension: 256.
  • Input length: 1024; resizing: resample.
  • Per-channel normalization: sample_minmax.
  • Pretraining-only heads and optimizer state are excluded. No fault-label classifier is included.
  • Source checkpoint, SHA256 and source code commit: see provenance.json.
  • All exported encoder tensors load strictly. Features and preprocessing were compared with the main-branch implementation (maximum feature error: 0).

Use the FreqFault repository:

python inference.py --model-id bojian1/UniFault-small --revision seed-42 --input samples.npy --output features.npz --seed 42

main defaults to training seed 42; seed-42, seed-43, and seed-44 select training checkpoints. The CLI --seed controls inference randomness, independently of the training seed. TF-ProFM retains random patch sampling; reproducibility requires the same seed, device, batch size and input order. UniFault returns mean-pooled token features.

Artifacts contain only inference configuration and safetensors weights; load using from model import load_model, UniFreqProcessor from FreqFault, not Transformers AutoModel.

Downloads last month
58
Safetensors
Model size
6.34M params
Tensor type
F32
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support