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

[models] Add binary classification model with pretrained weights #22

Closed
frgfm opened this issue Oct 15, 2019 · 0 comments
Closed

[models] Add binary classification model with pretrained weights #22

frgfm opened this issue Oct 15, 2019 · 0 comments
Assignees
Labels
help wanted Extra attention is needed module: models Related to models
Milestone

Comments

@frgfm
Copy link
Member

frgfm commented Oct 15, 2019

🚀 Feature

Create a models module in the package that allows users to pick a binary classification model with pretrained weights.

Motivation

While users are able to train on the dataset, for ease of use, a stable pretrained model should be accessible for people who only wish to perform inference.

Pitch

The module will implement model similarly to https://github.com/pytorch/vision/tree/master/torchvision/models where a pretrained argument can be set to True to load in the pretrained parameters.

@frgfm frgfm added the module: models Related to models label Oct 15, 2019
@frgfm frgfm added this to the 0.1.0 milestone Oct 15, 2019
@frgfm frgfm added the help wanted Extra attention is needed label Oct 15, 2019
@frgfm frgfm self-assigned this Oct 17, 2019
frgfm added a commit that referenced this issue Oct 22, 2019
Added a first resnet to the models module with a pretrained parameter loading option. Closes #22
@frgfm frgfm closed this as completed in 5989ee9 Oct 23, 2019
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
help wanted Extra attention is needed module: models Related to models
Projects
None yet
Development

No branches or pull requests

1 participant