Open
Description
Feature Request
The current version of scivision
only returns some features of the selected model. It'd be great to inform other relevant information such as number of parameters, model size, number of labels (for segmentation/classification models), etc. @darribas indicated about this feature as a feedback in a recent scivision
demo.
Describe the solution you'd like
Adding more info when describing the model e.g. size
scivision.PretrainedModel(
module='',
model='',
source=''
pipe='' ,
size= '',
parameters=''
)
Describe alternatives you've considered
Number of labels (classification) or range of values (regression) could be a tricky information to extract from the model. We could ask to model contributors to add this information somewhere in the scivision_model.yml
file?