This repository was archived by the owner on Sep 18, 2024. It is now read-only.
model space "num_labels" parameter inconsistent among various classes #5796
Description
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?