Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Pre-trained model loading API #277

Open
aaarrti opened this issue Jun 22, 2023 · 0 comments
Open

Pre-trained model loading API #277

aaarrti opened this issue Jun 22, 2023 · 0 comments
Assignees

Comments

@aaarrti
Copy link
Collaborator

aaarrti commented Jun 22, 2023

Description of the problem

Description of a solution

  • Provide API similar to torchvision or keras.application, e.g.
import quantus

model = quantus.sample_models.LeNet3D(weights="MNITS")

Which allows the user to directly execute examples.

  • Some weights are already out there, but probably we will need to train a few simple models, store in GitHub.
  • Weights, that can be downloaded from raw GitHub content under the hood and cached in local file system.
  • Provide clear description in readthedocs, which weights for which models are available.
@aaarrti aaarrti self-assigned this Jun 22, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant