refactor: simplify LeJEPA loss and module surface (#1926)
* refactor: update LeJEPA modules * fix: align LeJEPA tests and defaults * fix: use 1024 SIGReg vectors * fix: require LeJEPAProjectionHead input dim * test: drop LeJEPAProjectionHead type check * removed lejepa from rst * improve views documentation invariance loss * rename to proj_dim->output_dim to follow API * fix: harden LeJEPA loss API * fix: use output_dim for LeJEPA head tests * improve verification robustness * some differents amount of local vs global views tested * fix: use 3.8+ type hints * sigreg * validate projections once * tuple instead of Tuple * put shape validation on invariance loss
F
fruet committed
c7a06c22a479ff569be21e4fb94aea3e7e4b3bd1
Parent: 5aa9d51
Committed by GitHub <noreply@github.com>
on 5/19/2026, 2:15:47 PM