You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.
🚀 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 toTrue
to load in the pretrained parameters.The text was updated successfully, but these errors were encountered: