This repository has been archived by the owner on Sep 18, 2024. It is now read-only.
model space "num_labels" parameter inconsistent among various classes #5796
Open
Description
openedon Jul 3, 2024
when using the premade model spaces e.g. mobile net , there is a parameter to change the classification layer to align with our own dataset e.g.
Why is this not a parameter for other model spaces e.g. dart space :
for the Dart model space, there is only the option to use CIFAR or IMAGENET model spaces. But what if I want to change the number of labels to 2 for my own dataset?
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Metadata
Assignees
Labels
No labels