Mobilevit experiments: Retrieval

Overview

This is an experimental Mobilevit codebase for Retrieval. It keeps the xlarge setup intentionally manageable so architecture changes can be inspected before a full training run.

Default Experiment Recipe

Defaults: rmsprop optimizer, exponential schedule. Treat these as starting points; a real run should sweep learning rate and batch size before drawing conclusions.

Limitations

  • weights are random-init, so outputs carry no signal
  • robustness, fairness and domain transfer are untested
  • the API may change without notice while the design settles

Training Details

  • Weights initialized with a orthogonal scheme.
  • Runs on a single consumer GPU.
  • Full smoke test takes minutes, not hours.
  • No distributed setup required.

Changelog

  • 2026-10-13 โ€” initial upload
  • 2026-10-21 โ€” config and docs sync

Repository Status

  • The Python file contains the model and runnable example or training entry point.
  • config.json records the generated architecture settings.
  • training_args.json records the default experiment recipe.
  • model.safetensors is a valid initialization checkpoint for smoke tests; it is not presented as a trained benchmark checkpoint.
  • No benchmark score is claimed in this repository.

Files

  • inference.py โ€” primary artifact
  • README.md โ€” this documentation
  • config.json โ€” architecture configuration
  • training_args.json โ€” default experiment settings
  • model.safetensors โ€” initialization checkpoint

License

Released under bsd-3-clause. Review the source-data terms separately when this repository is used with external datasets.

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