search

paganpasta / eqxvision

A Python package of computer vision models for the Equinox ecosystem.
https://eqxvision.readthedocs.io
MIT License
100 stars 12 forks source link

Adding deserialization from torch saved weights #4

Closed paganpasta closed 2 years ago

paganpasta commented 2 years ago

An initial idea for the design is:

  1. Download weights from pre-hosted torchvision or timms URLs
  2. Have common converters which maps layers between the two frameworks. eg: torch.nn.Linear to eqx.nn.Linear
  3. Lastly, assembly of the weights can be performed by individual architectures.

Open to discussion!

paganpasta commented 2 years ago

torchvision pretrained support is now added!