Machine Learning

1 readers

1 users here now

Community Rules:

Be nice. No offensive behavior, insults or attacks: we encourage a diverse community in which members feel safe and have a voice.
Make your post clear and comprehensive: posts that lack insight or effort will be removed. (ex: questions which are easily googled)
Beginner or career related questions go elsewhere. This community is focused in discussion of research and new projects that advance the state-of-the-art.
Limit self-promotion. Comments and posts should be first and foremost about topics of interest to ML observers and practitioners. Limited self-promotion is tolerated, but the sub is not here as merely a source for free advertisement. Such posts will be removed at the discretion of the mods.

founded 1 year ago

MODERATORS

communick@academy.garden

Looking for a very small text classification transformer [R] (alien.top)

submitted 1 year ago by PresentDelivery4277@alien.top to c/machinelearning@academy.garden

5 comments fedilink hide all child comments

I'm looking for suggestions for a transformer model that I can fine-tune for a text classification task. Due to hardware constraints the model has to be fairly small. Something in the order of a 50 MB weight file.

you are viewing a single comment's thread
view the rest of the comments

[–] heisenbork4@alien.top 1 points 1 year ago

You could take a look here: https://sparsezoo.neuralmagic.com/?modelSet=natural_language_processing&size=10727959%2C64684665&sort=Size%3Aasc

Smallest model there is 10MB

I think it's still the case that it has to run on x86, though I think there was talk of an arm runtime.

Also, spending on your exact needs, there might be licensing issues. Still probably worth a look though