search

hezarai / hezar

The all-in-one AI library for Persian, supporting a wide variety of tasks and modalities!
https://hezarai.github.io/hezar/
Apache License 2.0
844 stars 45 forks source link

Add roberta for sequence labeling #65

Closed arxyzan closed 1 year ago

arxyzan commented 1 year ago

Add Roberta model from Transformers for sequence labeling. See BertSequenceLabeling and RobertaTextClassification for references.

pooya-mohammadi commented 1 year ago

@arxyzan jan, I'm on this

pooya-mohammadi commented 1 year ago

@arxyzan jan, please check it, if you want I can train it for more epochs. Since we want to save the logs, I think it's better to postpone the second training until after decision making of our logging mechanism

arxyzan commented 1 year ago

@pooya-mohammadi Can you send a PR?

pooya-mohammadi commented 1 year ago

https://github.com/hezarai/hezar/pull/73 @arxyzan jan,